上海计算机学会2024年6月月赛C++乙组T2幂次分解

幂次分解

内存限制: 256 Mb时间限制: 1000 ms

题目描述

给定两个正整数 n,m，你可以把 n 分解成若干个 m 的幂次方之和的形式，请你求出所有合法分解的方案数。（由于方案数可能非常多，输出方案数对 10^9+7 取模即可）

例如，当 n=6,m=2 时，可以有以下66种分解方式：
6=4+2
6=4+1+1
6=2+2+2
6=2+2+1+1
6=2+1+1+1+1
6=1+1+1+1+1+1

输入格式

输入两个正整数，分别表示 n,m

输出格式

输出一个正整数，表示分解方案数

数据范围

• 对于30%的数据：1≤n≤20，1≤m≤5
• 对于60%的数据：1≤n≤10^3，1≤m≤50
• 对于100%的数据：1≤n≤10^5，1≤m≤100

样例数据

输入:
6 2
输出:
6
输入:
10 3
输出:
5 

解析：

完全背包问题的方案数，详见代码：

#include <bits/stdc++.h>
using namespace std;
int n, m;
int dp[100005];
int main() {
    cin >> n >> m;
    dp[0] = 1;
    for(int i = 1; i <= n; i *= m) {
        for(int j = i; j <= n; j++) {
            dp[j] += dp[j - i];
            dp[j] %= 1000000007;
        }
    }
    cout << dp[n];
    return 0;
}