洛谷P2657 [SCOI2009]windy数(数位DP)

数位DP入门题,关于数位DP,可以参考下面几个博客:
https://walesexcitedmei.github.io/2018/10/30/浅浅浅谈-数位-DP/
https://www.luogu.org/blog/virus2017/shuweidp
https://blog.csdn.net/wust_zzwh/article/details/52100392
个人觉得第一二个会比较好懂一点。
代码如下:

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#include<iostream>
using namespace std;
const int maxn=1e3+5;
typedef long long ll;
int digit[20];
ll dp[20][20];	//dp[i,j]表示枚举到第i位时,它前一位(即i+1位)的数字为j时的方案数

ll DFS(int pos,int pre,bool lead,bool limit)	//pos表当前位数,pre表前一位的数字,lead标志是否前导0,limit标志是否有限制 
{
	if(pos==-1)	return 1;	//pos=-1的时候说明个位数是搜索过且符合条件,故返回1(类似于dp中的边界化处理) 
	if(!limit&&!lead&&dp[pos][pre]!=-1)	return dp[pos][pre];	//返回dp[pos][pre]的条件:未有限制;无前导0(已经开始数位DP);dp[pos][pre]有值 
	int up= limit? digit[pos]:9;	//up表数字上限
	ll res=0;
	for(int i=0;i<=up;i++)
	{
		int p=i;
		if(lead&&!i)	p=-10;
		if(abs(i-pre)<2)	continue;		//筛除不符合条件的数(类似于dp中的状态转移方程)
		res+=DFS(pos-1,p,p==-10,limit&&i==digit[pos]);
	 } 
	if(!limit&&!lead)	dp[pos][pre]=res;
	return res;
}

ll solve(ll n)
{
	int pos=0;
	while(n)
	{
		digit[pos++]=n%10;
		n/=10;
	}
	memset(dp,-1,sizeof(dp));//cout<<1<<endl;
	return DFS(--pos,-10,1,1);
}

int main()
{
	ll a,b;
	if(a>b) swap(a,b);
	cin>>a>>b;
	cout<<solve(b)-solve(a-1)<<endl;
	return 0;
}

小结:个人记得,数位DP的状态可以用DP的思维去定义,算法的话可以套模板(稍作修改)。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值