洛谷P2602 数字计数 数位dp

给出$1\le l,r\le 1e12$的区间，询问区间中$0,1,2...9$这些数位总共出现的次数。
考虑前缀作差$[1,r]-[1,l-1]$，然后从高位开始搜索，$dp[pos][sum][lim][zero]$表示搜索到$pos$位，而且当前这个数已经出现了$sum$次的答案。
注意到这里要记录$sum$是因为，前面搜索的次数不同对当前的答案也会产生影响。所以需要分开记录答案。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
ll dp[14][14][2][2];
int a[14]; 
ll dfs(int pos,int sum,bool lim,bool zero,int digit) {
	if(pos==0) return sum;
	if(dp[pos][sum][lim][zero]!=-1)
		return dp[pos][sum][lim][zero];
	int up=lim?a[pos]:9;
	ll ans=0;
	for(int i=0;i<=up;i++) {
		ans+=dfs(pos-1,sum+((i==digit)&&(i||!zero)),i==a[pos]&&lim,i==0&&zero,digit);
	}
	return dp[pos][sum][lim][zero]=ans;
}

ll solve(ll x,int digit) {
	if(x==0) return 0;
	memset(dp,-1,sizeof(dp));
	int len=0;
	while(x) {
		a[++len]=x%10;
		x/=10;
	}
	return dfs(len,0,1,1,digit);
}
int main() {
	ll l,r;
	scanf("%lld%lld",&l,&r);
	for(int i=0;i<10;i++)
		printf("%lld ",solve(r,i)-solve(l-1,i));
	return 0;
}