CSES-2220 Counting Numbers（数位 DP）

最新推荐文章于 2025-04-26 19:53:27 发布

edward1346

最新推荐文章于 2025-04-26 19:53:27 发布

阅读量297

点赞数 5

文章标签：算法动态规划 c++

本文链接：https://blog.csdn.net/2403_87021226/article/details/144564917

版权

题目传送门https://vjudge.net/problem/CSES-2220#author=GPT_zh这题目一看就像数位 DP，点进去一看果然是数位 DP。

解题思路

数位 DP 模板题……

转化成前缀和思想……

很容易想到设 $f(pos,last,lead,limit)$ 表示当前是从左往右的第 $pos$ 位，上一位是 $last$ ，当前是否有前导零，当前是否有最高位限制。

这里使用的是记忆化搜索。

代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
int l,r;
int len,num[21];
int f[21][11][2][2];
int dfs(int pos,int last,int lead,int limit)
{
	if(pos==0)
	{
		return 1;
	}
	if(f[pos][last][lead][limit]!=-1)
		return f[pos][last][lead][limit];
	int up=9;
	if(limit)
		up=num[pos];
	int ans=0;
	for(int i=0;i<=up;i++)
	{
		if(i==last)continue;
		if(lead&&i==0)
			ans+=dfs(pos-1,-1,1,limit&&i==up);
		else{
			ans+=dfs(pos-1,i,0,limit&&i==up);
		}
	}
	f[pos][last][lead][limit]=ans;
	return ans;
}
int solve(int x)
{
	len=0;
	while(x)
	{
		num[++len]=x%10;
		x/=10;
	}
	memset(f,-1,sizeof f);
	return dfs(len,-1,1,1);
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	cin>>l>>r;
	
	cout<<solve(r)-solve(l-1);
	return 0;
}