hdu 2089 数位dp

来源:互联网 发布:inode mac版 编辑:程序博客网 时间:2024/04/20 05:54

题意:给定一个区间[n,m],问在其中的不包含数字4,且不包含连续的62的数字有多少个?

思路:典型的数位dp。dp[i,j]表示开头为j的i位数中满足要求的数量。求和的时候先解析出数字,然后从高位往低位求和。

#include <cstdio>#include <cstring>#include <algorithm>#include <queue>#define clr(s,t) memset(s,t,sizeof(s));using namespace std;#define N 20005int dp[9][12],digit[9];int n,m;void init(){    clr(dp, 0);    dp[0][0] = 1;    for(int i = 1;i<=7;i++)        for(int j = 0;j<=9;j++)            for(int k = 0;k<=9;k++)                if(!(j==4 || ((j==6)&&(k==2))))                    dp[i][j] += dp[i-1][k];}int solve(int x){//[0,x-1]符合条件的个数    int len = 0;    int res = 0;    while(x){        digit[++len] = x%10;        x /= 10;    }    digit[len+1] = -1;    for(int i = len;i>=1;i--){        for(int j = 0;j<digit[i];j++)            if(!(j==4 || ((j==2)&&(digit[i+1]==6))))                res += dp[i][j];        if(digit[i] == 4 || ((digit[i]==2)&&(digit[i+1]==6)))//一个剪枝,当前位已经不满足了,不必往后求和啦            break;    }    return res;}int main(){    init();    while(scanf("%d %d",&n,&m) && (n+m))        printf("%d\n",solve(m+1)-solve(n));}


0 0