[ZJOI2010]数字计数 数位DP

题目描述

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

输入输出格式

输入格式：

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

输出格式：

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

最简单的数位DP的题目，可以对每个数字单独做，拿数字1举例。

考虑DP[x][high][limit][lead]，表示最高位为high的x位数中1出现的次数，其中limit表示该位是否受限，若受限，下一位的取值不能大于给定数字的第x-1位a[x-1]，lead表示是否有前导0

可以算出DP[x][high][limit][lead] = sum(DP[x-1][k][limit&&k == a[x-1]][lead && k == 0]) + cal(x,high,limit,lead)，若limit，k的取值从0到a[x-1]，否则k的取值从0到9。cal(x,high,limit,lead)计算了最高位high对答案的贡献，显然当high不为1的时候贡献为0，为1的时候计算结果为后x-1位在limit条件下的数的数量。

注意一下各种边界条件

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

long long dp[20][10][2][2];
long long pow10[] = {1,10,100,1000,10000,100000,1000000,10000000,1e8,1e9,1e10,1e11,1e12,1e13};
int a[20];

long long cal(int x,int high,bool limit,bool lead,int d){
	long long ans = 0;
	if(high != d || (lead && d == 0)) return 0;
	if(x == 0) return 1;
	if(!limit) return pow10[x];
	x--;
	while(x >= 0){
		ans *= 10;
		ans += a[x--];
	}
	return ans + 1;
}

long long solve(int x,int high,bool limit,bool lead,int d){
	if(dp[x][high][limit][lead] != -1) return dp[x][high][limit][lead];
	if(x == 0) return dp[x][high][limit][lead] = high == d ? 1 : 0;
	long long tmp = 0;
	if(limit){
		for(int k = 0;k <= a[x-1];k++)
			tmp += solve(x-1,k,k == a[x-1],lead && k == 0,d);
	} else{
		for(int k = 0;k < 10;k++)
			tmp += solve(x-1,k,false,lead && k== 0,d); 
	}
	tmp += cal(x,high,limit,lead,d);
	return dp[x][high][limit][lead] = tmp;
}

long long solve_problem(long long x,int d){
	if(x == 0) return d == 0 ? 1 : 0;
	int len;
	for(len = 0;x;len++){
		a[len] = x % 10;
		x /= 10;
	}
	memset(dp,-1,sizeof(dp));
	return solve(len,0,true,true,d);
}

int main(){
	long long l,r;
//	while(1){
		cin>>l>>r;
		for(int i = 0;i < 10;i++){
			cout<<(solve_problem(r,i)-solve_problem(l-1,i))<<" ";
		}
		cout<<endl;
//	}
}