HDU2089 不要62 数位DP

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2089


题目大意:给一个区间[n,m](1e7的范围),找出区间内不含62(连续的6和2)和4的数有多少个。


分析:数位DP,我们用f[i,st]来表示以st开头的符合题意的i位数的个数,对于每一次询问,分别统计[1,n]和[1,m+1]内的数,然后相减即可。


实现代码如下:

//数位DP,统计区间[n,m]内不含62和4的数的个数
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int M=1e7+2;
int f[8][10];
int ans[M+1];
void init()
{
    memset(f,0,sizeof(f));
    f[0][0]=1;
    for(int i=1;i<=7;i++) //1到7位数
      for(int j=0;j<=9;j++) //第i位由0到9
        for(int k=0;k<=9;k++) //第i+1位
          if( j!=4 && !(j==6&&k==2) )
            f[i][j]+=f[i-1][k];
}
int dig[10],len;  //从右到左纪录x的每一位
void change(int x)
{
    len=1;
    while(x)
    {
        dig[len++]=x%10;
        x/=10;
    }
    dig[len]=0;
}
int solve(int x)
{
    int ans=0;
    change(x);
    for(int i=len;i>=0;i--)
    {
        for(int k=0;k<dig[i];k++)
          if( k!=4 && !(k==2&&dig[i+1]==6) )
            ans+=f[i][k];
        if(dig[i]==4||(dig[i]==2&&dig[i+1]==6))
          break;
    }
    return ans;
}
int main()
{
    init();
    int n,m;
    while(scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0) break;
        if(n>m) swap(n,m);
        printf("%d\n",solve(m+1)-solve(n));
    }
    return 0;
}


你可能感兴趣的:(HDU2089 不要62 数位DP)