【数位DP】 hdu3555 Bomb

Bomb

题目:http://acm.hdu.edu.cn/showproblem.php?pid=3555

题意:给你一个数字N(1 <= N <= 2^63-1),问在1~N这些数字中,“49”这个序列出现多少次。

题解:数位DP。其实就是以一位数为单位的DP,有点类似AC自动机里的字典树。

           我们可以根据字典树把情况分为3种,序列中出现“49”,序列中没有“49”但是最后一位是“4”,序列中没有“49”且最后一位不为“4”。

           在这3种状态中进行转移即可。

         dp[k][0] = dp[k - 1][1] * 8 + dp[k - 1][0] * 9;      0 不含49且以非4的数字结尾
         dp[k][1] = dp[k - 1][0]  + dp[k - 1][1];                1 不含49且以4这个数字结尾
         dp[k][2] = dp[k - 1][2] * 10 + dp[k - 1][1];         2 含49的状态

代码:

#include<cstdio>
#include<cstring>
using namespace std;
#define LL long long
char s[25];
LL dp[25][3][2];
/*
dp[k][0] = dp[k - 1][1] * 8 + dp[k - 1][0] * 9;
dp[k][1] = dp[k - 1][0]  + dp[k - 1][1];
dp[k][2] = dp[k - 1][2] * 10 + dp[k - 1][1];
0 不含49且以非4的数字结尾
1 不含49且以4这个数字结尾
2 含49的状态
*/
int next(int a,int b)
{
    if(a==2||(a==1&&b==9)) return 2;
    else return b==4;
}
int main()
{
    int t;
    scanf("%d",&t);
    for(; t--;)
    {
        scanf("%s",s+1);
        memset(dp,0,sizeof(dp));
        int len=strlen(s+1);
        dp[0][0][1]=1;
        for(int i=1; i<=len; ++i)
        {
            dp[i][0][0] = dp[i-1][0][0]*9 + dp[i-1][1][0]*8;
            dp[i][1][0] = dp[i-1][0][0] + dp[i-1][1][0];
            dp[i][2][0] = dp[i-1][2][0] * 10 + dp[i-1][1][0];
            for(int j=0; j<s[i]-'0'; ++j)
            {
                dp[i][next(0,j)][0]+=dp[i-1][0][1];
                dp[i][next(1,j)][0]+=dp[i-1][1][1];
                dp[i][next(2,j)][0]+=dp[i-1][2][1];
            }
            dp[i][next(0,s[i]-'0')][1]+=dp[i-1][0][1];
            dp[i][next(1,s[i]-'0')][1]+=dp[i-1][1][1];
            dp[i][next(2,s[i]-'0')][1]+=dp[i-1][2][1];
        }
        printf("%I64d\n",dp[len][2][0]+dp[len][2][1]);
    }
}

来源: http://blog.csdn.net/ACM_Ted

你可能感兴趣的:(【数位DP】 hdu3555 Bomb)