数位dp一般解决给定区间【l,r】内满足某种条件的数的总数
例题1:https://www.luogu.org/problem/P2657
题意求出区间内的数满足相邻两位不小于2的总数
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 10 + 5;
ll dp[maxn][maxn]; //定义dp[i][j]表示第i位值为j时的方案数
vector<int> dim; //分解数字
//dfs定义为从高到低的第x位,前一个状态为st且前面的状态是否为边界
int dfs(int x, int st, int op) {
if (!x) return 1;
if (!op && ~dp[x][st]) return dp[x][st]; //记忆化搜索
int maxx = op ? dim[x] : 9, ret = 0;
for (int i = 0; i <= maxx; ++i) {
if (abs(st - i) < 2) continue;
if (st == 11 && i == 0)
ret += dfs(x - 1, 11, op & (i == maxx));
else
ret += dfs(x - 1, i, op & (i == maxx));
}
if (!op) dp[x][st] = ret;
return ret;
}
int solve(int x) {
memset(dp, -1, sizeof(dp));
dim.clear(); dim.push_back(-1);
while (x) {
dim.push_back(x % 10);
x /= 10;
}
return dfs(dim.size() - 1, 11, 1);
}
int main()
{
//freopen("in.txt", "r", stdin);
int l, r; scanf("%d%d", &l, &r);
printf("%d\n", solve(r) - solve(l - 1));
}
P2518 [HAOI2010]计数
求出由给出数字的重新排列后小于该数字的数目
即求给出数字的全排列中小于数字的结果
若第i位已经小于该数字的第i位,则后面的数字随便排列,即C(m,a[0])*C(m,a[1])*…*C(m,a[9]),a数组保存剩下的0-9的个数
#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ll;
const int maxn = 50 + 5;
char str[maxn];
int a[15], n;
ll ans, CC[100][100];
ll C(ll n, ll m) {
if (CC[n][m]) return CC[n][m];
if (m == 1) return n;
if (m == 0 || m == 1) return 1;
if (m > n) return 0;
CC[n][m] = C(n - 1, m) + C(n - 1, m - 1);
return CC[n][m];
}
ll cal() {
ll tmp = 1;
int m = n;
for (int i = 0; i < 9; ++i)
if (a[i]) tmp *= C(m, a[i]), m -= a[i];
ans += tmp;
}
int main()
{
freopen("in.txt", "r", stdin);
scanf("%s", &str); n = strlen(str);
for (int i = 0; i < n; ++i) {
a[str[i] - '0']++;
}
for (int i = 0, len = n; i < len; ++i) {
n--;
for (int j = 0; j < str[i] - '0'; ++j) {
if (a[j]) {
a[j]--; cal(); a[j]++;
}
}
a[str[i] - '0']--;
}
printf("%lld\n", ans);
}
P3413 SAC#1 - 萌数
求出给出区间【l,r】中满足长度大于2的回文数的数量
反向思考,当一个位置上的数与前两个位置的数都不相同时即不可能形成回文数
dp[i][j][k]表示当前i位置时,前面一个数为j,前前面一个数为k时的不能形成回文数的数量
注意处理前导0的情况,不能多算了
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9 + 7;
const int maxn = 1000 + 5;
ll dp[maxn][15][15];
int x[maxn], n;
ll dfs(int i, int j, int k, bool lim, bool zero) {
if (i == 0) return 1;
if (~dp[i][j][k] && ~j && ~k && !lim && !zero)
return dp[i][j][k];
int maxx = lim ? x[i] : 9;
ll res = 0;
for (int l = 0; l <= maxx; ++l) {
if (l != j && l != k && !zero)
res = (res + dfs(i - 1, l, j, lim && l == maxx, zero)) % mod;
else if (zero)
res = (res + dfs(i - 1, (l == 0 && zero) ? -1 : l, -1, lim && l == maxx, zero && l == 0)) % mod;
}
if (~j && ~k && !lim && !zero) dp[i][j][k] = res;
return res;
}
char l[maxn], r[maxn];
int main()
{
scanf("%s%s", &l, &r);
memset(dp, -1, sizeof(dp));
n = strlen(r);
ll cntl = 0, cntr = 0;
for (int i = 0; i < n; ++i) {
x[i + 1] = r[i] - '0';
cntr = (cntr * 10 % mod + x[i + 1]) % mod;
}
reverse(x + 1, x + n + 1);
ll ans = (cntr - dfs(n, -1, -1, 1, 1) + mod) % mod;
memset(dp, -1, sizeof(dp)); n = strlen(l);
for (int i = 0; i < n; ++i) {
x[i] = l[i] - '0';
cntl = (cntl * 10 % mod + x[i]) % mod;
}
cntl = (cntl - 1 + mod) % mod;
reverse(x, x + n); x[0] -= 1;
for (int i = 0; i < n; ++i) {
if (i == n - 1 && x[i] <= 0) {
n--;
break;
}
if (x[i] < 0) {
x[i] += 10;
x[i + 1] -= 1;
}
}
for (int i = n - 1; i >= 0; --i)
x[i + 1] = x[i];
if (!n) printf("%lld\n", ans);
else {
ans = (ans - (cntl - dfs(n, -1, -1, 1, 1) + mod) % mod + mod) % mod;
printf("%lld\n", ans);
}
}