数位DP的思想就是对每一位进行DP,计算时记忆化每一位可以有的状态,其作用是减少运算时间,避免重复计算。
数位DP是一种计数用的DP,一般就是要统计一个区间[A,B]内满足一些条件数的个数。
以1e9甚至1e18、1e100的问题为例,因为在统计情况下有很多重复的计算,数位DP实现了相同状态只计算一次,从而大幅减少运算时间。
数位DP:
技巧1:[X, Y] => f(Y) - f(X-1)
技巧2:用树进行排列
1.度的数量
1081
求给定区间[X, Y]中满足下列条件的整数个数:这个数恰好等于K个互不相等的B的整数次幂之和。
例如。设X=15,Y=20,K=2,B=2,择优且仅有下列三个数满足题意:
17 = 24 + 20
18 = 24 + 21
20 = 24 + 22
输入格式
第一行包含两个整数X和Y,接下来两行包含整数K和B
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 35;
int K, B;
int f[N][N];
void init()
{
for(int i = 0;i < N; i++)
for(int j = 0;j <= N; j++)
if(!j) f[i][j] = 1;
else f[i][j] = f[i - 1][j] + f[i - 1][j - 1];
}
int dp(int n)
{
if(!n) return 0;
vector<int> nums;
while(n)
{
nums.push_back(n % B);
n /= B;
}
int res = 0;
int last = 0;
for(int i = nums.size() - 1;i >= 0; i--)
{
int x = nums[i];
if(x)//求左边的分支
{
res += f[i][K - last];
if(x > 1)
{
if(K - last - 1 >= 0) res += f[i][K - last - 1];
break;
}
else
{
last++;
if(last > K) break;
}
}
if(!i && last == K) res++; // 最右侧分支上的方案
}
return res;
}
int main()
{
init();
int l, r;
cin >> l >> r >> K >> B;
cout << dp(r) - dp(r - 1) << endl;
return 0;
}
2.数字游戏
科协里最近很流行数字游戏。
某人命名了一种不降数,这种数字必须满足从左到右各位数字呈现非下降关系,如123,446
现在大家决定玩一个游戏,指定一个整数闭区间[a, b],问这个区间内有多少个不降数。
动态规划
-
状态表示
-
集合:所有最高位是j,且一共有i位的不降数的集合
-
属性:数量
-
-
状态计算
#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
const int N = 40;
int f[N][N]; // f[i][j]表示一共有i位,且最高位填j的数的个数
void init()
{
for(int i = 0;i <= 9; i++) f[1][i] = 1;
for(int i = 2;i < N; i++)
for(int j = 0;j <= 9; j++)
for(int k = j;k <= 9; k++)
f[i][j] += f[i - 1][k];
}
int dp(int n)
{
if(!n) return 1;
vector<int> nums;
while(n)
{
nums.push_back(n % 10);
n /= 10;
}
int res = 0;
int last = 0;
for(int i = nums.size() - 1;i >= 0; i--)
{
int x = nums[i];
for(int j = last;j < x; j++)
res += f[i + 1][j];
if(last > x)
break;
last = x;
if(!i) res++;
}
return res;
}
int main()
{
void init();
int l, r;
while(scanf("%d%d", &l, &r))
{
cout << dp[r] - dp[l - 1] << endl;
}
return 0;
}
3.Windy数
1083.Windy定义了一种Windy数:不包含前导零且相邻两个数字之差至少为2的正整数被称为Windy数。
Windy想知道,在A和B之间,包括A和B,总共有多少个Windy数?
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int N = 11;
int f[N][10];
void init()
{
for(int i = 0;i <= 9; i++) f[1][i] = 1;
for(int i = 2;i < N; i++)
for(int j = 0;j <= 9; j++)
for(int k = 0;k <= 9; k++)
if(abs(j - k) >= 2)
f[i][j] += f[i - 1][k];
}
int dp(int n)
{
if(!n) return 0;
vector<int> nums;
while(n) nums.push_back(n % 10), n /= 10;
int res = 0;
int last = -2;
for(int i = nums.size() - 1;i >= 0; i--)
{
int x = nums[i];
for(int j = 0;j < x; j++)
if(abs(j - last) >= 2)
res += f[i + 1][j];
if(abs(x - last) >= 2) last = x;
else break;
if(!i) res++;
}
// 特殊枚举有前导零
for(int i = 1;i < nums.size(); i++)
for(int j = 1;j <= 9;+)
res += f[i][j];
return res;
}
int main()
{
init();
int l, r;
cin >> l >> r;
cout << dp(r) - dp(l - 1) << endl;
return 0;
}
4.数字游戏II
1084.由于科协里最近真的很流行数字游戏。
某人又命名了一种取模数,这种数字必须满足各位数字之和mod N 为0
现在大家又要玩游戏了,指定一个整数闭区间[a, b],问这个区间内有多少个取模数。
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int N = 11, M = 110;
int P;
int f[N][10][M];
int mod(int x, int y)
{
return (x % y + y) % y;
}
void init()
{
memset(f, 0, sizeof(f));
for(int i = 0;i <= 9; i++) f[1][i][i % P]++;
for(int i = 2;i < N; i++)
for(int j = 0;j <= 9; j++)
for(int k = 0;k < N; k++)
for(int x = 0;x <= 9; x++)
f[i][j][k] += f[i - 1][x][mod(k - j, P)];
}
int dp(int n)
{
if(!n) return 1;
vector<int> nums;
while(n) nums.push_back(n % 10), n /= 10;
int res = 0;
int last = 0;
for(int i = nums.size() - 1;i >= 0; i--)
{
int x = nums[i];
for(int j = 0;j < x; j++)
res += f[i + 1][j][mod(-last, P)];
last += x;
if(!i && last % P == 0) res++;
}
return res;
}
int main()
{
int l, r;
while(cin >> l >> r >> P)
{
init();
cout << dp[r] - dp[l - 1] << endl;
}
return 0;
}
5.不要62
1085.杭州人称那些傻乎乎黏兮兮的人为62。
杭州交通管理局经常会扩充一些的士车牌照。附近出来一个好消息,以后上牌照,不再含有不吉利的数字了,这样一来,就可以消除个别的士机和乘客的心理障碍,更安全地服务大众。
不吉利的数字为所有含有4或62的号码。如:62315,73418,88914都属于不吉利号码。但是,61152虽然含有6和2,但不是逐号,所以不属于不吉利数字之列。
你的任务是,对于每次给出的一个牌照号区号[n, m],推断出交管局今后又要实际上给多少辆的士车上牌照了。
#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
const int N = 9;
int f[N][10];
void init()
{
for(int i = 0;i <= 9; i++)
if(i != 4)
f[1][i] = 1;
for(int i = 2;i < N; i++)
for(int j = 0;j <= 9; j++)
{
if(j == 4) continue;
for(int k = 0;k <= 9; k++)
{
if(k == 4 || j == 6 && k == 2)
continue;
f[i][j] += f[i - 1][k];
}
}
}
int dp(int n)
{
if(!n) return 1;
vector<int> nums;
while(n) nums.push_back(n % 10), n /= 10;
int res = 0;
int last = 0;
for(int i = nums.size() - 1;i >= 0; i--)
{
int x = nums[i];
for(int j = 0;j < x; j++)
{
if(j == 4 || last == 6 && j == 2) continue;
res += f[i + 1][j];
}
if(x == 4 || last == 6 && x == 2) break;
last = x;
if(!i) res++;
}
return res;
}
int main()
{
init();
int l, r;
while(cin >> l >> r, l || r)
{
cout << dp[r] - dp[l - 1] << endl;
}
return 0;
}
6.恨7不成妻
1086.DS级码农吉哥依然单身! 所以,他生平最恨情人节,不管是214还是77,他都讨厌!
吉哥观察了214和77这两个数,发现: 2+1+4=7 7+7=72 77=711 最终,他发现原来这一切归根到底都是因为和7有关!所以,他现在甚至讨厌一切和7有关的数!
什么样的数和7有关呢?
如果一个整数符合下面3个条件之一,那么我们就说这个整数和7有关—— 1、整数中某一位是7; 2、整数的每一位加起来的和是7的整数倍; 3、这个整数是7的整数倍;
现在问题来了:吉哥想知道在一定区间内和7无关的数字的平方和。
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
typedef long long LL;
const int N = 20, P = 1e9 + 7;
struct F
{
int s0, s1, s2;
}f[N][10][7][7];
int power7[N], power9[N];
int mod(LL x, int y)
{
return (x % y + y) % y;
}
void init()
{
for(int i = 0;i <= 9; i++)
{
if(i == 7) continue;
auto &v = f[1][i][i % 7][i % 7];
v.s0++;
v.s1 += i;
v.s2 += i * i;
}
LL power = 10;
for(int i = 2;i < N; i++, power *= 10)
for(int j = 0;j < N; j++)
{
if(j == 7) continue;
for(int a = 0;a < 7; a++)
for(int b = 0;b < 7; b++)
for(int k = 0;k <= 9; k++)
{
if(k == 7) continue;
auto &v1 = f[i][j][a][b], &v2 = f[i - 1][k][mod(a - j * (power % 7), 7)][mod(b - j, 7)];
v1.s0 = (v1.s0 + v2.s0) % P;
v1.s1 = (v1.s1 + j * (power % P) * v2.s0 + v2.s1) % P;
v1.s2 = (v1.s2 + j * j * (power % P) % P * (power % P) % P * v2.s0) % P + 2 * j * (power % P) % P * v2.s1 % P + v2.s2) % P;
}
}
power7[0] = power9[0] = 1;
for(int i = 1;i < N; i++)
{
power7[i] = power7[i - 1] * 10 % 7;
power9[i] = power9[i - 1] * 10 % P;
}
}
F get(int i, int j, int a, int b)
{
int s0 = 0, s1 = 0, s2 = 0;
for(int x = 0;x < 7; x++)
for(int y = 0;y < 7; y++)
{
if(x == a || y == b) continue;
auto v = f[i][j][x][y];
s0 = (s0 + v.s0) % P;
s1 = (s1 + v.s1) % P;
s2 = (s2 + v.s2) % P;
}
return {s0, s1, s2};
}
int dp(LL n)
{
if(!n) return 0;
LL backup_n = n % P;
vector<int> nums;
while(n) nums.push_back(n % 10), n /= 10;
int res = 0;
LL last_a = 0, last_b = 0;
for(int i = nums.size() - 1;i >= 0; i--)
{
int x = nums[i];
for(int j = 0;j < x; j++)
{
if(j == 7) continue;
int a = mod(-last_a % 7 * power7[i + 1], 7);
int b = mod(-last_b, 7);
auto v = get(i + 1, j, a, b);
res = (res + (last_a % P) * (last_a * P) % P * power9[i + 1] % P * power9[i + 1] % P * v.s0 % P + 2 (last_a % P) % P * power9[i + 1] % P * v.s1 % P + v.s2) % P;
}
if( x == 7) break;
last_a = last_a * 10 + x;
last_b += x;
if(!i && last_a % 7 && last_b % 7) res += (res + backup_n * backup_n) % P;
}
return res;
}
int main()
{
init();
int T;
cin >> T;
while(T--)
{
int l, r;
cin >> l >> r;
cout << mod(dp(r) - dp(l - 1), P) << endl;
}
return 0;
}