数位DP 第一题 hdu 2089

来源：互联网发布：知廉耻懂敬畏编辑：程序博客网时间：2024/05/20 04:26

不要62 和4

转载于 http://blog.csdn.net/acm_cxlove/article/details/7819907

#include<iostream>#include<cstring>#include<queue>#include<cstdio>#include<cmath>#include<algorithm>#define N 55#define inf 1<<29#define MOD 9973#define LL long long#define eps 1e-7#define zero(a) fabs(a)<eps#define equal(a,b) zero(a-b)using namespace std;int dp[10][3];//dp[i][0],表示不存在不吉利数字//dp[i][1],表示不存在不吉利数字，且最高位为2//dp[i][2],表示存在不吉利数字void Init(){    memset(dp,0,sizeof(dp));    dp[0][0]=1;    for(int i=1;i<=6;i++){        dp[i][0]=dp[i-1][0]*9-dp[i-1][1];  //在最高位加上除了4之外的9个数字，但是可能在2之前加了6        dp[i][1]=dp[i-1][0];    //就是在原先不含不吉利数字的最高位加2        dp[i][2]=dp[i-1][2]*10+dp[i-1][0]+dp[i-1][1];  //在已经有不吉利数字最高位加任意数字，或者在无吉利数字前加4，或者在2前面加4    }}int slove(int n){    int len=0,bit[10];    int tmp=n;    while(n){        bit[++len]=n%10;        n/=10;    }    bit[len+1]=0;    int ans=0;    bool flag=false;    for(int i=len;i;i--){        ans+=dp[i-1][2]*bit[i];          if(flag)   //高位已经出现4或者62，后面的就随意            ans+=dp[i-1][0]*bit[i];        if(!flag&&bit[i]>4)  //高位可能出现4的情况            ans+=dp[i-1][0];        if(!flag&&bit[i+1]==6&&bit[i]>2)  //高位是6，后面一位可能出现2，这步debug了很久            ans+=dp[i][1];        if(!flag&&bit[i]>6)  //高位可能出现6，要把后面最高位为2计入            ans+=dp[i-1][1];        if(bit[i]==4||(bit[i+1]==6&&bit[i]==2))  //高位已经出现4或者62            flag=true;    }    return tmp-ans;}int main(){    int l,r;    Init();    while(scanf("%d%d",&l,&r)!=EOF&&l+r)        printf("%d\n",slove(r+1)-slove(l));    return 0;}

还有数位dp的 dfs写法：

　　dp[i][0]:前i位不含不吉利数的个数。

　　dp[i][1]:前i位不含不吉利数且i+1位是6的个数。

　　dp[i][2]:前i位含不吉利数的个数。

#include<cstdio>#include<cstring>#include<algorithm>#define N 10using namespace std;int bit[N];int dp[N][3];int dfs(int pos,int st,bool flag){    if(pos==0)return st==2;    if(flag&&dp[pos][st]!=-1)return dp[pos][st];    int ans=0;    int u=flag?9:bit[pos];    for(int d=0;d<=u;d++){        if(st==2||d==4||(st==1&&d==2))ans+=dfs(pos-1,2,flag||d<u);        else if(d==6)ans+=dfs(pos-1,1,flag||d<u);        else ans+=dfs(pos-1,0,flag||d<u);    }    if(flag)dp[pos][st]=ans;    return ans;}int solve(int n){    int len=0;    while(n){        bit[++len]=n%10;        n/=10;    }    return dfs(len,0,0);}int main(){    int n,m;    memset(dp,-1,sizeof(dp));    while(~scanf("%d%d",&n,&m)){        if(!(n||m))return 0;        printf("%d\n",m-n+1-(solve(m)-solve(n-1)));    }    return 0;}