Harmony Pairs（数位 dp）

最新推荐文章于 2021-06-08 13:20:48 发布

幽影相随

最新推荐文章于 2021-06-08 13:20:48 发布

阅读量388

点赞数

分类专栏： # 动态规划（DP）

本文链接：https://blog.csdn.net/weixin_43820352/article/details/107633476

版权

动态规划（DP）专栏收录该内容

64 篇文章 1 订阅

订阅专栏

Harmony Pairs

题目描述

Roundgod is obsessive about numbers. Let $S (x)$ be the sum of the digits of xx in decimal notation, $(A, B)$ is a harmony pair if and only if $S (A) > S (B)$ . Roundgod is given $N$ , and she wants to count the number of harmony pairs $(A, B)$ modulo 10⁹ + 7 satisfying $0 \leq A \leq B \leq N$ .

输入描述:

The only line of input contains one integer N (1 ≤ N ≤ 10¹⁰⁰).

输出描述:

Output one integer indicating the answer.

输入

100

输出

967

题意

给一个 $N$ 满足几个条件， $1 ≤ N ≤ 10^{100})$

$S (A) > S (B)$
$1 < A < B < N$
在 $1$ ~ $n$ 中存在几组不同的 $A ， B$

#include<bits/stdc++.h>

#define ll long long
using namespace std;
const int mod = 1e9 + 7;
ll dp[110][2020][3][3];
char s[210];
int bit[210];
int n;

ll dfs(int pos, int sum, bool flaga, bool flagb) {
    if (pos < 0) {
        return sum > 1000;
    }
    if (dp[pos][sum][flaga][flagb] != -1) {
        return dp[pos][sum][flaga][flagb];
    }
    ll ans = 0;
    int aend = flaga ? bit[pos] : 9;
    // a
    for (int i = 0; i <= aend; i++) {
        // b
        int bend = flagb ? i : 9;
        for (int j = 0; j <= bend; j++) {
            ans = (ans + dfs(pos - 1,
                             sum - i + j,
                             flaga && (i == aend),
                             flagb && (j == bend)) % mod) % mod;
        }
    }
    return dp[pos][sum][flaga][flagb] = ans;
}

int main() {
    scanf("%s", s);
    n = strlen(s);
    for (int i = 0; i < n; i++) {
        bit[n - i - 1] = s[i] - '0';
    }
    memset(dp, -1, sizeof(dp));
    printf("%lld\n", dfs(n - 1, 1000, 1, 1));
    return 0;
}