Dp练习 -Sakura数简单数位DP

Dave_L

于 2018-06-29 20:35:44 发布

阅读量255

点赞数

CC 4.0 BY-SA版权

分类专栏： DP 动态规划-数位DP

本文为博主原创文章，未经博主允许也可以转载。

本文链接：https://blog.csdn.net/FrankAx/article/details/80861594

DP 同时被 2 个专栏收录

21 篇文章

订阅专栏

动态规划-数位DP

14 篇文章

订阅专栏

本文介绍了一种简单的数位动态规划方法，通过示例代码详细解释了如何利用DP解决特定问题，包括处理前导0及数据溢出的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这里写图片描述

思路：简单数位DP，dp[pos][pre]表示枚举到pos位时前一位为pre。
考虑前导0以及数据范围自然溢出。

Code：

#include <bits/stdc++.h>
#define LL unsigned long long
using namespace std;
const int AX = 20+2;
int a[AX];
LL dp[AX][15];

LL dfs( int pos , int pre , bool lead , bool lim ){
    if( pos == -1 ) return 1;
    if( !lim && !lead && dp[pos][pre] != -1 ) return dp[pos][pre];
    LL ans = 0;
    int up = lim ? a[pos] : 9 ;
    for( int i = 0 ; i <= up ; i++ ){
        if( pre == -1 || abs( pre - i ) >= 2 ){
            if( lead && i == 0 ) ans += dfs( pos - 1 , pre , lead, lim && i == a[pos]);
            else ans += dfs( pos - 1 , i , lead && i == 0 , lim && i == a[pos] );
        }
    }
    if( !lim && !lead ) dp[pos][pre] = ans;
    return ans ;
}

LL solve( LL x ){   
    int tot = 0 ;
    while( x ){
        a[tot++] = x % 10;
        x /= 10;
    }
    return dfs( tot - 1 , -1 ,  true , true );
}

int main(){
    LL l , r;
    cin >> l >> r;
    memset( dp , -1 , sizeof(dp) );
    //printf("%lld\n",solve(r)-solve(l-1));
    cout << solve(r) - solve(l-1);
    return 0 ;
}