hdu 2089 不要62 (數位DP)

題意:給定一個區間[n,m],求n到m中沒有“62”和“4”的數的個數。其中例如62548 ,412包含62 和4 所以是不合法的。

思路:設dp[i][j]表示i位數且開頭爲j的數中不包含“4”和“62"的數的個數,則可以得出dp[i][j]=dp[i][j]+dp[i-1][k],(0=<k<=9),則可以將0000000~9999999的數進行一下預處理。然後對n,m分別從高位開始統計,統計到比n的小的數字,算出[0,m)和[0,n)的合法個數,最後對m進行特殊判斷即可,答案即爲[0.m)-[0,n),如果m合法則答案加1.

代碼如下:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int f[10][12];
int numn[10];//存儲n的第i位的數字是幾,從右往左存
int numm[10];//存儲m的第i位的數字是幾,從右往左存
int lenn;//n的位數
int lenm;//m的位數
int n,m;
void work()//0000000~9999999的預處理
{
    memset(f,0,sizeof(f));
    f[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)))
                {
                     f[i][j]+=f[i-1][k];
                }
            }
}
void solve(int n,int m)//對n,m進行分離每位上的數字
{
    lenn=0,lenm=0;
    while(n)
    {
        numn[++lenn]=n%10;
        n/=10;
    }
    while(m)
    {
        numm[++lenm]=m%10;
        m/=10;
    }
}
int check()//判斷n是否合法
{
    for(int i=lenm;i>0;i--)
    {
        if(numm[i]==4) return 0;
        if((i<lenm)&&(numm[i+1]==6)&&(numm[i]==2)) return 0;
    }
    return 1;
}
int cal()//計算[n,m]中合法的個數
{
    int sum=0;
    int res=0;
    for(int i=lenn;i>0;i--)
    {
        for(int j=0;j<numn[i];j++)
        {
            if(j!=4)
            {
               if((i<lenn)&&(j==2)&&(numn[i+1]==6)) continue;
                    sum+=f[i][j];
            }
        }
        if(numn[i]==4)
        {
            break;
        }
        if((i<lenn)&&(numn[i+1]==6)&&(numn[i]==2))
        {
            break;
        }
    }
    for(int i=lenm;i>0;i--)
    {
        for(int j=0;j<numm[i];j++)
        {
            if(j!=4)
            {
                if((i<lenm)&&(j==2)&&(numm[i+1]==6)) continue;
                    res+=f[i][j];
            }
        }
        if(numm[i]==4)
        {
            break;
        }
        if((i<lenm)&&(numm[i+1]==6)&&(numm[i]==2))
        {
            break;
        }
    }
    return res-sum;
}
int main()
{
   // freopen("in.txt","r",stdin);
    work();
    while(scanf("%d%d",&n,&m),n||m)
    {
        solve(n,m);
        printf("%d\n",cal()+check());
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章