hdu3555,hdu2089 数位dp

第一次做数位dp，最开始的2089题我是用暴力过得，根本没想去想数位dp，这几天刚好学到这，就返回去又做了一遍，这两道题都算是入门吧，好像还有一种递归的写法，准备学一下。

hdu 3555

#include<stdio.h>
#include<string.h>
//#include<iostream>
//using namespace std;
long long  dp[20][5],m=0;
int dig[20];

int main()
{
    int N;
    memset(dp,0,sizeof(dp));
    dp[0][0]=1;
    for(int i=1; i<20; i++)
    {
        dp[i][0]=dp[i-1][0]*10-dp[i-1][1];
        dp[i][1]=dp[i-1][0];
        dp[i][2]=dp[i-1][2]*10+dp[i-1][1];
    }
    scanf("%d",&N);
    while(N--)
    {

        int cnt=1;
        memset(dig,0,sizeof(dig));
         long long ans=0;
         scanf("%I64d",&m);
         m++;
        while(m)
        {
            dig[cnt++]=m%10;
            m/=10;
        }

        int flag=0;
        for(int i=cnt-1; i>=1; i--)
        {
            ans+=dig[i]*dp[i-1][2];
            if(flag)
            {
                ans+=dig[i]*dp[i-1][0];
            }
            else
            {
                if(dig[i]>4) ans+=dp[i-1][1];
            }
            if(dig[i+1]==4&&dig[i]==9)
                flag=1;
        }
        printf("%I64d\n",ans);
    }
    return 0;

}

hdu2089

#include<stdio.h>
#include<string.h>
int dp[10][5],dig[15];
int solve(int t)
{
    int i=0,temp=t,ans=0;
  //  memset(dig,0,sizeof(dig));
    while(temp)
    {
        dig[++i]=temp%10;
        temp/=10;
    }
    dig[i+1]=0;
    int flag=0;
    for(int cnt=i;cnt>0;cnt--)
    {
        ans+=dp[cnt-1][2]*dig[cnt];
        if(flag)
        {
            ans+=dig[cnt]*dp[cnt-1][0];
        }
        else
        {
            if(dig[cnt]>4) ans+=dp[cnt-1][0];
            if(dig[cnt]>6) ans+=dp[cnt-1][1];
            if(dig[cnt+1]==6&&dig[cnt]>2)
                ans+=dp[cnt][1];
        }
        if(dig[cnt]==4||(dig[cnt+1]==6&&dig[cnt]==2) )
            flag=1;
    }
    return t-ans;

}
int main()
{
    int n,m,i,k,t,temp,temp1,x,y;
    memset(dp,0,sizeof(dp));
    //memset(dp2,0,sizeof(dp2));
   // dp[1][0]=9;
    dp[0][0]=1;
    for(i=1;i<=8;i++)
    {

        dp[i][0]=dp[i-1][0]*9-dp[i-1][1];
       // else
            dp[i][1]=dp[i-1][0];
        dp[i][2]=dp[i-1][2]*10+dp[i-1][1]+dp[i-1][0];
        //dp2[i][0]=(dp2[i-1][0]+dp2[i-1][1])*9;
    }

    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0&&m==0) break;
        k=solve(m+1)-solve(n);
        printf("%d\n",k);
    }
    return 0;
}

好吧，这智商也是醉了，想了好几天递归的写法，终于看得差不多了，其实就是上面过程的相反，反过来算，突然就发现原来不难。

hdu2089

#include<stdio.h>
#include<string.h>
int n,m;
int dp[50][5],dig[50];
int dfs(int wei,int sta,int flag)
{
    int l=sta,ans1=0;
    if(wei==0) return sta==2;
    if(dp[wei][sta]!=-1&&!flag) return dp[wei][sta];
    int len=flag? dig[wei]:9;
    for(int i=0;i<=len ;i++)
    {
        l=sta;
        if(i!=4&&i!=6&&sta==1&&i!=2) l=0;
        if(i==6&&sta!=2) l=1;
        if(sta==1&&i==2) l=2;
        if(i==4) l=2;
        ans1+=dfs(wei-1,l,flag&&(i==len));
    }
    if(!flag) dp[wei][sta]=ans1;
    return ans1;
}
int solve(int x)
{
    int pos=1;
    while(x)
    {
        dig[pos++]=x%10;
        x/=10;
    }
    dig[pos]=0;
    return dfs(pos-1,0,1);
}
int main()
{
    int ans;
    memset(dp,-1,sizeof(dp));
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0&&m==0) break;
        int temp=m-n+1;
      //  printf("%d\n",solve(n-1));
        ans=solve(m)-solve(n-1);
        printf("%d\n",temp-ans);
    }
    return 0;
}

hdu3555

#include<stdio.h>
#include<string.h>
__int64 n,ans;
__int64 dp[20][5],dig[20];
__int64 dfs(int wei,int sta,int flag)//wei 记录位数，sta记录状态，flag 记录前一位是否最大
{
    __int64 ans1=0;
    int l;
    //printf("%d\n",wei);
    if(wei==0) return sta==2;
    //printf(" %d  %d\n",wei,flag);
    if(dp[wei][sta]!=-1&&flag==0) return dp[wei][sta];
    int len=flag? dig[wei]:9;
    for(int i=0;i<=len;i++)
    {
        l=sta;
        if(sta==1&&i==9) l=2;
        if(sta==0&&i==4) l=1;
        if(sta==1&&i!=4&&i!=9) l=0;
        ans1+=dfs(wei-1,l,flag&&(i==len));
    }
    if(flag==0)
    dp[wei][sta]=ans1;
    return ans1;
}
__int64 solve(__int64 x)
{
     int pos = 1;
    while (n > 0) {

        dig[pos] = n % 10;
        n /= 10,pos++;
    }
    return dfs(pos-1,0,1);
}
int main()
{
    int t;

    scanf("%d",&t);
    while(t--)
    {
        memset(dp,-1,sizeof(dp));
        scanf("%I64d",&n);
      //  n++;
        ans=solve(n);
        printf("%I64d\n",ans);
    }
    return 0;
}