【BZOJ1833】数字计数(ZJOI2010)-数位DP

测试地址:数字计数
做法:本题需要用到数位DP。
首先套路把答案转化为前缀和的形式(ans(r)ans(l1))。对于计算前n个数对答案的贡献,考虑计算每个数字在每一位上的贡献,即有多少个数在第i位上是j。分情况讨论:
1.若前i1位不卡上界,那么当前位一定可以选到j,所以满足要求的数字有:前i1位不卡上界的数字组合数×10种。
2.若前i1位卡上界,但当前位填j的话不卡上界,满足要求的数字有:10种。
3.若前i1位卡上界,而当前位填j的话也卡上界,满足要求的数字有:后面所有位不超过上界的数字组合数种。
注意前导0不对答案做出任何贡献,当前面i1位的上界是0时,当前位不能填0,而当前位如果填0,计算前i1位的数字组合时一定要减去全为0的情况。
以下是本人代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxdigit=13;
ll a,b,ans[31]={0},s[31];
ll f[31],g[31],up[31],ten[31];

void calc(ll type)
{
    f[0]=g[0]=0;
    for(int i=1;i<=maxdigit;i++)
    {
        g[i]=g[i-1]*10+9;
        if (i>1) ten[i]=ten[i-1]*10;
        else ten[i]=1;
        f[i]=f[i-1]+s[i]*ten[i];
    }
    up[maxdigit+1]=0;
    for(int i=maxdigit;i>=1;i--)
        up[i]=up[i+1]*10+s[i];
    for(int i=0;i<=maxdigit;i++)
        g[i]++,f[i]++;
    for(int i=maxdigit;i>=1;i--)
    {
        for(int j=0;j<=9;j++)
        {
            if (j==0)
            {
                if (up[i+1]>0||s[i]>0)
                {
                    ans[j]+=type*(up[i+1]-1)*g[i-1];
                    if (j<s[i]) ans[j]+=type*g[i-1];
                    else if (j==s[i]) ans[j]+=type*f[i-1];
                }
            }
            else
            {
                ans[j]+=type*up[i+1]*g[i-1];
                if (j<s[i]) ans[j]+=type*g[i-1];
                else if (j==s[i]) ans[j]+=type*f[i-1];
            }
        }
    }
}

int main()
{
    scanf("%lld%lld",&a,&b);

    a--;
    for(int i=1;i<=maxdigit;i++)
        s[i]=a%10,a/=10;
    calc(-1);

    for(int i=1;i<=maxdigit;i++)
        s[i]=b%10,b/=10;
    calc(1);

    for(int i=0;i<=9;i++)
        printf("%lld ",ans[i]);

    return 0;
}
posted @ 2018-04-16 21:12  Maxwei_wzj  阅读(108)  评论(0编辑  收藏  举报