HDU 3555 数位DP

第一道数位DP哦。。。

有关数字的东西真是打脑壳。。

不过真是个好东西，复杂度降低了不是一点一点，从网上的神牛哪里学到的。。

下面转自哪里不清楚的：

DP的状态是2维的dp[len][3]
dp[len][0] 代表长度为len不含49的方案数
dp[len][1] 代表长度为len不含49但是以9开头的数字的方案数
dp[len][2] 代表长度为len含有49的方案数

状态转移如下
dp[i][0] = dp[i-1][0] * 10 - dp[i-1][1];  // not include 49  如果不含49且，在前面可以填上0-9但是要减去dp[i-1][1]因为4会和9构成49
dp[i][1] = dp[i-1][0];  // not include 49 but starts with 9  这个直接在不含49的数上填个9就行了
dp[i][2] = dp[i-1][2] * 10 + dp[i-1][1]; // include 49  已经含有49的数可以填0-9，或者9开头的填4

接着就是从高位开始统计

在统计到某一位的时候，加上dp[i-1][2] * digit[i] 是显然对的，因为这一位可以填 0 - (digit[i]-1)
若这一位之前挨着49，那么加上dp[i-1][0] * digit[i] 也是显然对的。
若这一位之前没有挨着49，但是digit[i]比4大，那么当这一位填4的时候，就得加上dp[i-1][1]

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<stdlib.h>
#include<math.h>
#include<vector>
using namespace std;
#define maxn 1010
#define ll __int64
typedef long long LL;
ll dp[20][3];
int digit[20];
ll n;
//dp[i][0] 长度为i包含49的个数
//dp[i][1] 长度为i不包含49，但是以9开头
//dp[i][2] 长度为i不包含49
void init()
{
    memset(dp,0,sizeof(dp));
    dp[0][2]=1;
    for(int i=1;i<20;i++)
    {
        dp[i][0]=(ll)(dp[i-1][0]*10)+dp[i-1][1];
        dp[i][1]=dp[i-1][2];
        dp[i][2]=(ll)(dp[i-1][2]*10)-dp[i-1][1];
    }
}
void DP()
{
    scanf("%I64d",&n);
    n++;
    memset(digit,0,sizeof(digit));
    int len=0;
    ll ans=0;
    int flag=0,last=0;
    while(n)
    {
        digit[++len]=n%10;
        n/=10;
    }
   // cout<<len<<endl;
    for(int i=len;i>=1;i--)
    {
        ans+=dp[i-1][0]*digit[i];
        if(flag)
        ans+=dp[i-1][2]*digit[i];
        if(!flag&&digit[i]>4)
        ans+=dp[i-1][1];
        if(last==4&&digit[i]==9)
        flag=1;
        last=digit[i];
    }
    printf("%I64d\n",ans);
}
int main()
{
    freopen("in.txt","r",stdin);
    int t;
    scanf("%d",&t);
    init();
    while(t--)
    {
        DP();
    }
    return 0;
}