数位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的思维去定义,算法的话可以套模板(稍作修改)。