hdu 2089 数位dp

题意:给定一个区间[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 20005
int 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));
}


你可能感兴趣的:(hdu 2089 数位dp)