简介
数位dp一般解决的是对于一个区间 [ l,r ],求出满足要求的数的个数,所谓数位dp,就是对每个位的数进行dp。
数位dp的实质就是换一种暴力枚举的方式,使得新的枚举方式满足dp的性质,然后记忆化就可以了。
例题
不要62
就是数位上不能有4也不能有连续的62,没有4的话在枚举的时候判断一下,不枚举4就可以保证状态合法了,所以这个约束没有记忆化的必要,而对于62的话,涉及到两位,当前一位是6或者不是6这两种不同情况我计数是不相同的,所以要用状态来记录不同的方案数。
dp[pos][sta]表示当前第pos位,前一位是否是6的状态,这里sta只需要去0和1两种状态就可以了,不是6的情况可视为同种,不会影响计数。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[20];
int dp[20][2];
int dfs(int pos,int pre,int sta,bool limit)
{//dp到的位数,前一位,前一位是否为6,是否到达边界
if(pos==-1) return 1;//设置递归出口
if(!limit && dp[pos][sta]!=-1) return dp[pos][sta];
//如果有记录且不是边界直接返回
int up=limit ? a[pos] : 9;
//判断是否达到边界
int tmp=0;
for(int i=0;i<=up;i++)
{
if(pre==6 && i==2)continue;
if(i==4) continue;//都是保证枚举合法性
tmp+=dfs(pos-1,i,i==6,limit && i==a[pos]);
//limit是判断前面是否都达到了边界
//i==a[pos]判断这一位是否达到边界
//如果都达到边界那么下一位上限就是a[pos]
}
if(!limit) dp[pos][sta]=tmp;//记忆化
return tmp;
}
int solve(int x)
{
int pos=0;
while(x)//把数的各个数位扔进数组
{
a[pos++]=x%10;
x/=10;
}
return dfs(pos-1,-1,0,true);
}
int main()
{
int le,ri;
//memset(dp,-1,sizeof dp);可优化
while(~scanf("%d%d",&le,&ri) && le+ri)
{
memset(dp,-1,sizeof dp);
printf("%d\n",solve(ri)-solve(le-1));
//solve(ri)表示的是1到ri的数量
//所以求区间的数量只要相减就行
}
return 0;
}
这里有个优化,可以吧memset(dp,-1,sizeof dp);放在外面,因为题目的要求没有变,只有输入的数据再变化,所以记忆化过的数据可以再次利用。