数位DP

hdu 3555

多组数据,问从1~n中有多少个数包含"49"

注意:数位DP最后统计答案的时候,*b[i]的计数表示0~b[i]-1种选法,统计的是比b[i]少1开头的数字之后有多少种,不会重复计算。

就像一棵十叉数,n+1在某一个叶子节点上,每次统计一层左子树的答案,然后下一层,再累和,所以从高位到低位循环。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
using namespace std;
#define LL long long
LL n,ans;
int T,len;
bool flag;
LL dp[30][3];
int b[30];
void pre()
{
    memset(dp,0,sizeof(dp));
    dp[0][0]=1;//边界很重要
    for (int i=1;i<=21;i++)
    {
        dp[i][0]=(LL)dp[i-1][0]*10-dp[i-1][1];
        dp[i][1]=dp[i-1][0];
        dp[i][2]=(LL)dp[i-1][2]*10+dp[i-1][1];
    }
}
int main()
{
    cin>>T;
    pre();//统一预处理
    while (T--)
    {
        cin>>n;
        n++;//统计的是[0,n-1]
        len=0;
        while (n)
        {
            b[++len]=n % 10;
            n/=10;
        }
        ans=0;
        flag=false;b[len+1]=0;//多组数据
        for (int i=len;i;i--)
        {
            ans+=(LL)dp[i-1][2]*b[i];
            if (flag)
                ans+=(LL)dp[i-1][0]*b[i];
            else
                if (b[i]>4)
                 ans+=dp[i-1][1];
            if (b[i]==9 && b[i+1]==4) flag=true;
        }
        cout<<ans<<endl;
    }
	return 0;
}


你可能感兴趣的:(数位dp)