BZOJ1833[count 数字计数]

题目描述

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

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

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

1 99

样例输出

9 20 20 20 20 20 20 20 20 20

提示

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

题解：数位DP，用dp[pos][sum][num]来表示在第pos位，目标是num，个数是sum时的值

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
/**************************************************************
    Problem: 1833
    User: Venishel
    Language: C++
    Result: 正确
    Time:40 ms
    Memory:1352 kb
****************************************************************/

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#define ll long long
using namespace std;
ll dp[20][20][20],a[20],l,r;

ll dfs(int pos,int num,ll sum,bool lim,bool flg)
{
    if(pos==0)return sum;
    if(!flg && !lim && dp[pos][num][sum]!=-1)return dp[pos][num][sum];
    int up=flg?a[pos]:9;
    ll ans=0;
    if(!lim||pos==1)ans+=dfs(pos-1,num,sum+(num==0),0,flg && a[pos]==0);
    else ans+=dfs(pos-1,num,sum,1,flg && a[pos]==0);
    for(int i=1;i<=up;i++)
    ans+=dfs(pos-1,num,sum+(num==i),0,flg && a[pos]==i);
    if(!flg && !lim)dp[pos][num][sum]=ans;
    return ans;
}
ll solve(ll x,int num)
{
    if(x<0)return 0;
    if(x==0)return num==0?1:0;
    int pos=0;
    while(x)
    {
        a[++pos]=x%10;
        x/=10;
    }
    return dfs(pos,num,0,1,1);
}
int main()
{
    cin >> l >> r;
    memset(dp,-1,sizeof(dp));
    for(int i=0;i<=8;i++)
     cout << solve(r,i)-solve(l-1,i)<< " ";
     cout << solve(r,9)-solve(l-1,9);
}