数位动态规划常用语解决动态规划中的计数类问题(计数类问题就是求方案数的问题),具体,一般会给定一个区间[x,y],让你求这个区间里满足某种性质的数字的个数。
从这个问题出发,给定区间[x,y],让我们求区间中满足某种性质的数字的个数,最简单的暴力想法是,枚举区间中的每一个数,逐个判断,确定满足条件的数字的个数,当数据范围较大时,这种方法显然会超时,因此我们引入数位dp。
一, 数位dp的基本思想
1, 数位,顾名思义,就是数字的,每一位,例如,个位,十位,百位…
2, 数位的dp的思想是换一种枚举的方式,从最高位开始枚举,这就有了像dp一样的决策阶段,每一位便是每一个阶段,数位dp常常用记忆化搜索来实现。
3,数位dp在统计方案数的时候用到了差分的思想,统计[n,m]区间里符合要求的数字的个数,常常统计前n-1个数符合条件的数的个数,再统计前m个数符合条件的数的个数,最后相减即是答案。
二, 数位dp的基本模板
(转自dalaohttps://blog.csdn.net/wust_zzwh/article/details/52100392)
typedef long long ll;
int a[20];
ll dp[20][state];//不同题目状态不同
ll dfs(int pos,/*state变量*/,bool lead/*前导零*/,bool limit/*数位上界变量*/)//不是每个题都要判断前导零
{
//递归边界,既然是按位枚举,最低位是0,那么pos==-1说明这个数我枚举完了
if(pos==-1) return 1;/*这里一般返回1,表示你枚举的这个数是合法的,那么这里就需要你在枚举时必须每一位都要满足题目条件,也就是说当前枚举到pos位,一定要保证前面已经枚举的数位是合法的。不过具体题目不同或者写法不同的话不一定要返回1 */
//第二个就是记忆化(在此前可能不同题目还能有一些剪枝)
if(!limit && !lead && dp[pos][state]!=-1) return dp[pos][state];
/*常规写法都是在没有限制的条件记忆化,这里与下面记录状态是对应,具体为什么是有条件的记忆化后面会讲*/
int up=limit?a[pos]:9;//根据limit判断枚举的上界up;这个的例子前面用213讲过了
ll ans=0;
//开始计数
for(int i=0;i<=up;i++)//枚举,然后把不同情况的个数加到ans就可以了
{
if() ...
else if()...
ans+=dfs(pos-1,/*状态转移*/,lead && i==0,limit && i==a[pos]) //最后两个变量传参都是这样写的
/*这里还算比较灵活,不过做几个题就觉得这里也是套路了
大概就是说,我当前数位枚举的数是i,然后根据题目的约束条件分类讨论
去计算不同情况下的个数,还有要根据state变量来保证i的合法性,比如题目
要求数位上不能有62连续出现,那么就是state就是要保存前一位pre,然后分类,
前一位如果是6那么这意味就不能是2,这里一定要保存枚举的这个数是合法*/
}
//计算完,记录状态
if(!limit && !lead) dp[pos][state]=ans;
/*这里对应上面的记忆化,在一定条件下时记录,保证一致性,当然如果约束条件不需要考虑lead,这里就是lead就完全不用考虑了*/
return ans;
}
ll solve(ll x)
{
int pos=0;
while(x)//把数位都分解出来
{
a[pos++]=x%10;//个人老是喜欢编号为[0,pos),看不惯的就按自己习惯来,反正注意数位边界就行
x/=10;
}
return dfs(pos-1/*从最高位开始枚举*/,/*一系列状态 */,true,true);//刚开始最高位都是有限制并且有前导零的,显然比最高位还要高的一位视为0嘛
}
int main()
{
ll le,ri;
while(~scanf("%lld%lld",&le,&ri))
{
//初始化dp数组为-1,这里还有更加优美的优化,后面讲
printf("%lld\n",solve(ri)-solve(le-1));
}
}
三, 数位dp的入门例题
1,(HDU2089)不要62
杭州人称那些傻乎乎粘嗒嗒的人为62(音:laoer)。
杭州交通管理局经常会扩充一些的士车牌照,新近出来一个好消息,以后上牌照,不再含有不吉利的数字了,这样一来,就可以消除个别的士司机和乘客的心理障碍,更安全地服务大众。
不吉利的数字为所有含有4或62的号码。例如:
62315 73418 88914
都属于不吉利号码。但是,61152虽然含有6和2,但不是62连号,所以不属于不吉利数字之列。
你的任务是,对于每次给出的一个牌照区间号,推断出交管局今次又要实际上给多少辆新的士车上牌照了。
分析:题目中的限制条件是,不能出现连续的62或者4,所以我们只要在枚举的时候记录一下前驱,如果当前枚举到4就舍弃,如果某个位置的前驱是6,当前节点枚举到2就舍弃,同时我们要开两维的dp数组记录最优值,第一维记录当前枚举到的位数,第二位记录当前点的前驱是否是6,因为6会对之后的节点的排列产生影响,比如一个节点是6,那么它的下一个节点就不能填2,否则可以,方案数有差别,会影响记忆化。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int dp[20][2],num[20];
inline int dfs(int pos,int pre,int sta,bool limit)
{
if(pos==0) return 1;
if(!limit&&dp[pos][sta]!=-1) return dp[pos][sta];
int up,sum=0;
if(!limit) up=9; else up=num[pos];
for(int i=0;i<=up;i++)
{
if(i==4) continue;
if(pre==6&&i==2) continue;
sum+=dfs(pos-1,i,i==6,limit&&(i==num[pos]));
}
if(!limit) dp[pos][sta]=sum;
return sum;
}
inline int solve(int k)
{
int pos=0;
while(k)
{
num[++pos]=k%10;
k/=10;
}
return dfs(pos,0,0,true);
}
int main()
{
int m,n;
memset(dp,-1,sizeof(dp));
while(1)
{
scanf("%d%d",&m,&n);
if(m==0&&n==0) return 0;
printf("%d\n",solve(n)-solve(m-1));
}
return 0;
}