【BZOJ-1833】count数字计数 数位DP

1833: [ZJOI2010]count 数字计数

Time Limit: 3 Sec  Memory Limit: 64 MB
Submit:
2494  Solved: 1101
[Submit][Status][Discuss]

Description

给定两个正整数a和b，求在[a,b]中的所有整数中，每个数码(digit)各出现了多少次。

Input

输入文件中仅包含一行两个整数a、b，含义如上所述。

Output

输出文件中包含一行10个整数，分别表示0-9在[a,b]中出现了多少次。

Sample Input

1 99

Sample Output

9 20 20 20 20 20 20 20 20
20

HINT

30%的数据中，a<=b<=10^6；
100%的数据中，a<=b<=10^12。

Source

Day1

Solution

裸的数位DP，但其实并不是特别的水

首先F[i][j][k]表示位数为i的最高位为j的k种数的个数

按照十进制拆分，预处理后统计答案

Code

#include<iostream>

#include<cstdio>

#include<algorithm>

#include<cstring>

#include<cmath>

using namespace std;

long long L,R;

long long F[][][],ans1[],ans2[];

void prework()

{

    for (int i=; i<=; i++) F[][i][i]=;

    long long tmp=;

    for (int i=; i<=; i++)

        {

            tmp*=;

            F[i][][]=F[i-][][]*+F[i-][][]+tmp;

            for (int j=; j<=; j++)

                F[i][][j]=F[i-][][j]*+F[i-][j][j];

            for (int j=; j<=; j++)

                {

                    F[i][j][]=F[i-][][]*+F[i-][][];

                    for (int k=; k<=; k++)

                        if (j==k)

                            F[i][j][k]=F[i-][][k]*+F[i-][k][k]+tmp;

                        else

                            F[i][j][k]=F[i-][][k]*+F[i-][k][k];

                }

        }

}

long long cf(int x)

{

    long long re=;

    for (int i=; i<x; i++)

        re*=;

    return re;

}

void Calc(long long x,long long *ans)

{

    int digit[]={},len=; long long y=x;

    while (x) {digit[++len]=x%; x/=;}

    for (int i=; i<len; i++)

        for (int j=; j<=; j++)

            for (int k=; k<=; k++)

                ans[k]+=F[i][j][k];

    for (int i=len; i>=; i--)

        {

            for (int j=; j<=digit[i]-; j++)

                {

                    if (i==len && j==) continue;

                    for (int k=; k<=; k++) ans[k]+=F[i][j][k];

                }

            ans[digit[i]]+=y%cf(i)+;

        }

}

int main()

{

    prework();

    scanf("%lld%lld",&L,&R);

    Calc(L-,ans1); Calc(R,ans2);

    printf("%lld",ans2[]-ans1[]);

    for (int i=; i<=; i++) printf(" %lld",ans2[i]-ans1[i]);

    return ;

}

自己一开始YY的出错了..