[BZOJ1799][Ahoi2009]self 同类分布(数位DP)

本文介绍了一种解决特定区间内数字拆分问题的动态规划算法。通过将[a,b]区间内的数字拆分为两个子区间[1,a-1]和[1,b],并利用记忆化搜索优化状态转移过程,实现高效求解。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

询问[a,b][a,b]拆成[1,a1][1,a−1][1,b][1,b]
状态定义:
f[i][j][k][h][0/1]f[i][j][k][h][0/1]表示:
到了第ii位,各位数的和为j,整个数模kkh,第五维为00表示不卡a1bb的上界,否则卡上界。
然后枚举第i位数的数值,大力转移即可。
这样空间是爆掉的,但可以发现一个性质:
转移的过程中,kk的值是不变的。
因此可以先枚举k,每次做一遍DP,f[i][j][h][0/1]f[i][j][h][0/1]同上。
可使用记忆化搜索,去掉不可能的状态,提高时间效率。
代码:

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define For(i, a, b) for (i = a; i <= b; i++)
using namespace std;
typedef long long ll;
const int N = 27, M = 178;
int n, a[N], times, mark[N][M][M][2], pw[M][M];
ll f[N][M][M][2];
ll dp(int x, int sm, int md, int op, int MX)
{
    if (sm < 0) return 0;
    if (x == 1)
    {
        mark[x][sm][md][op] = times;
        if (sm > 9) return f[x][sm][md][op] = 0;
        if (sm % MX != md) return f[x][sm][md][op] = 0;
        if ((!op && sm > a[1]) || (op && sm <= a[1]))
            return f[x][sm][md][op] = 0;
        return f[x][sm][md][op] = 1;
    }
    if (mark[x][sm][md][op] == times)
        return f[x][sm][md][op];
    int i; ll ans = 0;
    For (i, (op ? a[x] : 0), (op ? 9 : a[x]))
        if (i != a[x]) ans += dp(x - 1, sm - i,
            (md - pw[MX][x - 1] * i % MX + MX) % MX, 0, MX) +
            dp(x - 1, sm - i,
            (md - pw[MX][x - 1] * i % MX + MX) % MX, 1, MX);
        else ans += dp(x - 1, sm - i,
            (md - pw[MX][x - 1] * i % MX + MX) % MX, op, MX);
    return mark[x][sm][md][op] = times, f[x][sm][md][op] = ans;
}
ll solve(ll x)
{
    int i;
    n = 0;
    if (x == 0) return 0;
    while (x) a[++n] = x % 10, x /= 10;
    ll ans = 0;
    For (i, 1, 162)
        times++, ans += dp(n, i, 0, 0, i);
    return ans;
}
int main()
{
    int MX, j;
    For (MX, 1, 162)
    {
        pw[MX][0] = 1 % MX;
        For (j, 1, 162)
            pw[MX][j] = pw[MX][j - 1] * 10 % MX;
    }
    ll A, B;
    cin >> A >> B;
    cout << solve(B) - solve(A - 1) << endl;
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值