数位DP YbtOJ 数字计数

数位DP YbtOJ 数字计数_第1张图片

每一个数都是独立的,所以分别考虑0到9

首先询问l到r,很容易想到一个前缀和的思想,前r个减去前l-1个,就是区间l到r

考虑传参,pos,limit,lead,dig,sum,分别表示位数,填数范围的限制,是否有前导零,当前考虑的是哪个,出现次数。

然后就是数位DP的版子

#include
#define int long long
using namespace std;
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch == '-') f=-1 ; ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48) ; ch=getchar();}
	return x*f;
}
const int M = 20;
int dp[M][M],num[M];
int a,b;
int dfs(int pos,int limit,int lead,int dig,int sum){ //lead=0表示有前导0
	int ans=0;
	if(!pos) return sum;
	if(!limit && lead && dp[pos][sum] != -1) return dp[pos][sum]; //可以记忆化
	int res = limit ? num[pos] : 9;
	for(register int i(0) ; i<=res ; i=-~i) ans+=dfs(pos-1, (i==res)&&limit , (lead||i) , dig , sum+((i|| lead)&&(i==dig))); //记录答案,接着搜
	if(!limit && lead) dp[pos][sum] = ans; //可以存起来
	return ans;
}
int calc(int x,int d){
	int cnt=0;
	memset(dp,-1,sizeof(dp));
	while(x) num[++cnt]=x%10,x/=10;
	return dfs(cnt,1,0,d,0);
}
signed main(){
	a=read();b=read();
	for(register int i(0) ; i<=9 ; i=-~i) printf("%lld ",calc(b,i)-calc(a-1,i));
	return 0;
}

你可能感兴趣的:(蓝桥杯,职场和发展,动态规划,c++,算法)