【数位dp】【HDU 3555】【HDU 2089】数位DP入门题

【HDU  3555】原题直通车：

代码：

// 31MS 900K 909 B G++



#include<iostream>

#include<cstdio>

#include<cstring>

using namespace std;

int a[20];

__int64 dp[20][11];

void digit_dp() {

    memset(dp, 0LL, sizeof(dp));

    dp[0][0]=1;

    for(int i=1; i<20; ++i)

        for(int j=0; j<10; ++j)

            for(int k=0; k<10; ++k) {

                if(j==4&&k==9) continue;

                dp[i][j]+=dp[i-1][k];

            }

}

__int64 Count(__int64 x) {

    __int64 u=x, ret=0LL;

    int t=0;

    memset(a, 0, sizeof(a));

    while(u) a[++t]=u%10, u/=10;

    for(int i=t; i>=1; --i) {

        for(int j=0; j<a[i]; ++j) ret+=dp[i][j];

        if(a[i]==9&&a[i+1]==4){

            ret--; break;

        }

    }

    return ret;

}

int main() {

    int T; scanf("%d",&T);

    digit_dp();

    while(T--) {

        __int64 n;

        scanf("%I64d",&n);

        __int64 ans=Count(n);

        printf("%I64d\n",n-ans);

    }

    return 0;

}

 

 

 

 

【HDU  2089】原题直通车：

代码：

// 15MS 844K 921 B C++



#include<iostream>

#include<cstdio>

#include<cstring>

using namespace std;

int dp[10][10];

void digit_dp() {

    memset(dp, 0, sizeof(dp));

    dp[0][0]=1;

    for(int i=1; i<8; ++i)

        for(int j=0; j<10; ++j)

            for(int k=0; k<10; ++k) {

                if(j==4||k==4||(j==6&&k==2)) continue;

                dp[i][j]+=dp[i-1][k];

            }

}

int a[10];

int Count(int x) {

    memset(a, 0, sizeof(a));

    int u=x, ret=0, t=0;

    while(u) a[++t]=u%10, u/=10;

    for(int i=t; i>=1; --i) {

        for(int j=0; j<a[i]; ++j){

            ret+=dp[i][j];

        }

        if(a[i+1]==6){

            if(a[i]==2) break;

            if(a[i]>2)ret-=dp[i][2];

        }

        if(a[i]==4) break;

    }

    return ret;

}

int main() {

    int n, m;

    digit_dp();

    while(~scanf("%d%d",&n,&m)&&n+m)

        printf("%d\n",Count(m+1)-Count(n));

    return 0;

}