HDOJ 3555 Bomb （按位DP）

题目大意：求1到n中有多少个数含“49”

第一次接触按位DP，参考了别人的题解。

第一个代码运行速度快些（31ms），但状态的转移的分析比较麻烦，第二个代码要慢些（125ms），但code比较方便。

#include <stdio.h>
#include <string.h>
#define LEN 22
typedef __int64 LL;
char s[LEN];
LL dp[LEN][3][2];
int next(int cur,int in)
{
    if(cur==2 || cur==1&&in==9) return 2;
    return in==4;
}
LL DP(char s[],int len)
{
    memset(dp,0,sizeof(dp));
    dp[0][0][1]=1;
    for(int i=1;i<=len;i++)
    {
        dp[i][0][0]=9*dp[i-1][0][0]+8*dp[i-1][1][0];
        dp[i][1][0]=dp[i-1][0][0]+dp[i-1][1][0];
        dp[i][2][0]=10*dp[i-1][2][0]+dp[i-1][1][0];

        for(int j=0;j<s[i]-'0';j++)
        {
            dp[i][next(0,j)][0]+=dp[i-1][0][1];
            dp[i][next(1,j)][0]+=dp[i-1][1][1];
            dp[i][next(2,j)][0]+=dp[i-1][2][1];
        }

        dp[i][next(0,s[i]-'0')][1]+=dp[i-1][0][1];
        dp[i][next(1,s[i]-'0')][1]+=dp[i-1][1][1];
        dp[i][next(2,s[i]-'0')][1]+=dp[i-1][2][1];
    }
    return dp[len][2][0]+dp[len][2][1];
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",s+1);
        printf("%I64d\n",DP(s,strlen(s+1)));
    }
    return 0;
}

#include <stdio.h>
#include <string.h>
#define LEN 22
typedef __int64 LL;
char s[LEN];
LL dp[LEN][3][2];
int nextstate(int cur,int in)
{
    if(cur==2 || cur==1&&in==9) return 2;
    return in==4;
}
int nextflag(int cur,int in,int max)
{
    if(cur==1 && in==max)   return 1;
    if(cur==1 && in>max)    return -1;
    return 0;
}
LL DP(char s[],int len)
{
    memset(dp,0,sizeof(dp));
    dp[0][0][1]=1;
    for(int i=1;i<=len;i++)
    {
        for(int state=0;state<3;state++)
        {
            for(int flag=0;flag<2;flag++)
            {
                for(int j=0;j<10;j++)
                {
                    if(nextflag(flag,j,s[i]-'0')==-1)   continue;
                    dp[i][nextstate(state,j)][nextflag(flag,j,s[i]-'0')]+=dp[i-1][state][flag];
                }
            }
        }
    }
    return dp[len][2][0]+dp[len][2][1];
}
int main()
{
    freopen("in.txt","r",stdin);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",s+1);
        printf("%I64d\n",DP(s,strlen(s+1)));
    }
    return 0;
}