数位dp

数位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);
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值