[BZOJ1833] [ZJOI2010]count 数字计数

[ZJOI2010]count 数字计数

Description
给定两个正整数a和b，求在[a,b]中的所有整数中，每个数码(digit)各出现了多少次。
Input
输入文件中仅包含一行两个整数a、b，含义如上所述。
Output
输出文件中包含一行10个整数，分别表示0−9在[a,b]中出现了多少次。
Sample Input
199
Sample Output
9202020202020202020
HINT
30%的数据中，a<=b<=106；
100%的数据中，a<=b<=1012。

Solution :
记dp[i][less]表示做到第i位，是否贴着上限时0 9的个数以及此时一共有多少个数，我们先暴力枚举第一个非零位，然后dfs。

Code :

/*************************************************************************    > File Name: bzoj1833.cpp    > Author: Archer ************************************************************************/#include <bits/stdc++.h>#include <ext/pb_ds/assoc_container.hpp>#include <ext/pb_ds/hash_policy.hpp>using namespace std;const int N = 22;typedef long long ll;#define MS(_) memset(_, 0xff, sizeof(_))struct info{    ll ans[N], cnt;    info() {memset(ans, 0, sizeof(ans)); cnt = 0;}}dp[N][2];char str[N];int vis[N][2];ll L, R;info operator +(const info &x, const info &y){    info z; for (int i = 0; i <= N; i++) z.ans[i] = x.ans[i] + y.ans[i]; return z;}inline info dfs(int dep, int less){    info ret, pack;    if (~vis[dep][less]) { vis[dep][less] = 1; return dp[dep][less]; }    vis[dep][less] = 1;    if (dep >= strlen(str)){ ret.cnt = 1; dp[dep][less] = ret; return ret; }    for (int i = 0; i <= 9; i++)        if (less || i <= str[dep] - '0'){            pack = dfs(dep + 1, (less || i < str[dep] - '0') ? 1 : 0);            ret = ret + pack; ret.ans[i] += pack.cnt;         }    return dp[dep][less] = ret;}int main(){    scanf("%lld%lld", &L, &R); L--; MS(vis);    sprintf(str, "%lld", L); info Lans, pack;     for (int i = 0; i < strlen(str); i++)        for (int j = 1; j <= (i == 0 ? str[0]-'0' : 9); j++){            info pack = dfs(i + 1, (i == 0 && j == str[0] - '0') ? 0 : 1);            Lans = Lans + pack; Lans.ans[j] += pack.cnt;        }    sprintf(str, "%lld", R); info Rans; MS(vis);    for (int i = 0; i < strlen(str); i++)        for (int j = 1; j <= (i == 0 ? str[0]-'0' : 9); j++){            info pack = dfs(i + 1, (i == 0 && j == str[0] - '0') ? 0 : 1);            Rans = Rans + pack; Rans.ans[j] += pack.cnt;        }    for (int i = 0; i < 9; i++) printf("%lld ", Rans.ans[i] - Lans.ans[i]);     printf("%lld\n", Rans.ans[9] - Lans.ans[9]);    return 0;}