数位dp的实质:
- 一种快速枚举的方式,它满足dp的性质,然后进行记忆化搜。
例题 : 牛客-数位小孩
题目描述:
九峰最近沉迷数位dp,这天他又造了个数位dp题:
给出一个区间[l,r][l,r][l,r],求这个区间内有多少个数字满足如下条件:
1.每相邻两个数位和为素数。
2.其中至少一个数位为1。
3.没有前导0。
请你编写程序帮他该题。
输入描述:
一行两个整数l,r(1 <= l <= r <= ),表示给出的区间为[l,r]。
输出描述:
一行一个整数,表示区间内满足条件的个数。
思路:
大体思路:我们计算1 ~ r中间满足条件的个数减去1 ~ l - 1 满足条件的个数。
- 首先我们先进行预处理,处理出第i位,最高为j的所有满足条件条件的数的个数。
- 然后对l或r的每一位从最高位到最低位进行枚举(假设x为每一位需要枚举的数),切记因为不能包含前导零所以最高位要从1开始枚举到x - 1,其他的位数就是从0到x - 1。
- 最后我们每次都要判断l或r的位数中是否有含有相邻的两个数相加不是质数如果有枚举完那一位之后就直接跳出循环,因为继续枚举后面的的数都不会满足任意相邻的两个数相加为质数。如果走到了最后还要记得+1因为l或r也满足。
前导零问题:
- 是否需要特殊处理前导0这个需要根据前导0的存在是否会影响题目的性质。
- 这个题目会影响比如说如果你把 0012 和 12 看成一样的就会有影响,因为在0012 中0 + 0 == 0不是质数所以不符合,而 12 中1 + 2 == 3符合。
- 在AcWing - 1082.数字游戏 中是没有影响的因为 0012 和 12 都是不降数,0不会对后面有影响。
AC代码:
#include<iostream>
#include<set>
#include<vector>
using namespace std;
# define int long long
const int N = 15;
set<int>se;
int a[8] = {2,3,5,7,11,13,17,19};
int f[N][N][2];
void init()
{
for(int i = 0;i < 8;i ++) se.insert(a[i]);
for(int i = 0;i <= 9;i ++) f[1][i][i == 1] = 1;
for(int i = 2;i < 15;i ++)
{
for(int j = 0;j < 10;j ++)
{
for(int k = 0;k < 10;k ++)
{
if(se.count(j + k))
{
if(j == 1)
f[i][j][1] += f[i - 1][k][0] + f[i - 1][k][1];
else
{
f[i][j][1] += f[i - 1][k][1];
f[i][j][0] += f[i - 1][k][0];
}
}
}
}
}
}
int dp(int n)
{
if(!n) return 0;
vector<int>ve;
while(n)
{
ve.push_back(n % 10);
n /= 10;
}
int res = 0, last = -100, flag = 0;
n = ve.size() - 1;
for(int i = ve.size() - 1;i >= 0;i --)
{
int x = ve[i];
for(int j = i == n;j < x;j ++)
{
if(!se.count(last + j) && last > 0) continue;
if(j == 1 || flag)
res += (f[i + 1][j][0] + f[i + 1][j][1]);
else
res += f[i + 1][j][1];
}
if(last != -100 && !se.count(last + x)) break;
if(x == 1)flag = 1;
last = x;
if(!i && flag) res ++;
}
for(int i = 1;i < ve.size();i ++)
{
for(int j = 1;j < 10;j ++)
{
res += f[i][j][1];
}
}
return res;
}
signed main()
{
init();
int l,r;
cin >> l >> r;
cout << dp(r) - dp(l - 1) << endl;
return 0;
}
预处理:
我觉得这一步是最难想出来的
const int N = 15;
set<int>se;
int a[8] = {2,3,5,7,11,13,17,19};
int f[N][N][2];//三个括号分别表示最高位数,最高位所表示的值,是否包含1
void init()
{
for(int i = 0;i < 8;i ++) se.insert(a[i]);
for(int i = 0;i <= 9;i ++) f[1][i][i == 1] = 1;
for(int i = 2;i < 15;i ++)
{
for(int j = 0;j < 10;j ++)//j表示本位
{
for(int k = 0;k < 10;k ++)//k表示上一位
{
if(se.count(j + k))//判断相邻的两位是否位质数
{
if(j == 1)//当最高位是1时无论前面的位数中是否包含1都是可以算进来的
f[i][j][1] += f[i - 1][k][0] + f[i - 1][k][1];
else
{
//这里只能根据前面是否包含1来把它分成包含1和不包含1的两部分
f[i][j][1] += f[i - 1][k][1];
f[i][j][0] += f[i - 1][k][0];
}
}
}
}
}
}
dp函数:
int dp(int n)
{
if(!n) return 0;
vector<int>ve;
while(n)
{
ve.push_back(n % 10);
n /= 10;
}
int res = 0, last = -100;//last用于记录上一位,last == -100是为了便于第一次循环进行特判
int flag = 0;//flag是为了方便进行判断n的数位中是否包含有1
n = ve.size() - 1;
for(int i = ve.size() - 1;i >= 0;i --)
{
int x = ve[i];
for(int j = i == n;j < x;j ++)//i == n小技巧
{
if(!se.count(last + j) && last > 0) continue;
if(j == 1 || flag)
res += (f[i + 1][j][0] + f[i + 1][j][1]);
else
res += f[i + 1][j][1];
}
if(last != -100 && !se.count(last + x)) break;
if(x == 1)flag = 1;
last = x;
if(!i && flag) res ++;
}
//特殊处理前导0
for(int i = 1;i < ve.size();i ++)
{
for(int j = 1;j < 10;j ++)
{
res += f[i][j][1];
}
}
return res;
}
好了,这就是我最近学数位dp的一些心得,希望可以与大家一起分享。