通常我们会遇到这样的一类问题:给定一个上下限,求这个区间内满足某个条件的数的个数。
一般最简单的解法就是暴力枚举了:
for(int i = low;i <= up;i++)
{
if(right(condition)) count++;
}//low为下限,up为上限,right(condition)为给定的条件
但是往往区间会开的很大,这时候就会超时,数位dp就是更为快捷的解决这类问题;
例如
奖券数目
有些人很迷信数字,比如带“4”的数字,认为和“死”谐音,就觉得不吉利。
虽然这些说法纯属无稽之谈,但有时还要迎合大众的需求。某抽奖活动的奖券号码是5位数(10000-99999),要求其中不要出现带“4”的号码,主办单位请你计算一下,如果任何两张奖券不重号,最多可发出奖券多少张。
请提交该数字(一个整数),不要写任何多余的内容或说明性文字。
结果是52488;
先看暴力解法:
#include<iostream>
#include<string>
#include<sstream>
using namespace std;
int main()
{
int count = 0;
for(int i = 10000; i <= 99999;i++)
{
stringstream ss;
ss << i;
string s = ss.str();
if(s.find('4')==string::npos) count++;
}
cout << count;
return 0;
}
用了0. 0.09087秒。
再看数位dp的解法:
#include<iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
const int up = 99999,low = 10000;
int f[5];
int x[5][2];
int dfs(int len,bool lim)
{
if(len == -1) return 1;
int top = lim?f[len]:9;
int ans = 0;
for(int i = 0;i <= top;i++){
if(i==4)continue;
else{
ans+=dfs(len-1,lim&&i==top);
}
}
return ans;
}
int solve(int p)
{
int i = 0;
while(p)
{
f[i++] = p%10;
p/=10;
}
return dfs(4,1);
}
int main()
{
cout << solve(up) - solve(low)+1;
return 0;
}
用了0.01546;
快了近6倍,这里运行的时候看起来基本都是秒出的,是因为数据比较小,当数据一大,期间的差距就很显然了;
当然只是简单的数位dp,只判断了一个数,往往是判断两位数或者某个数的倍数的,那时就要用到记忆化了,这才是真正的dp;