【Codeforces Round #247 (Div. 2) C】k-Tree

题目链接

链接

翻译

给你一个每个节点都有 \(k\) 个孩子的无限树，和第 \(i\) 个孩子的连边的权值为 \(i\)。
让你找路径数，这个路径要满足以下条件:

• 从根开始
• 路径的权值和为n
• 至少有一条边的权值大于等于d

问你路径数目。

题解

动态规划
设 \(f[i][j]\) 表示权值和为 \(i\), 当前路径中是否出现大于等于 \(d\) 的权值。( \(0\) 表示没出现 \(1\) 表示出现了)。

\(f[i][0]\) 能转移到 \(f[i][1]\)。其他的不难。

代码

#include <bits/stdc++.h>
using namespace std;

/*
    f[i][j]

    f[i][0] 表示权重和为i,权重都小于d
    f[i][1] 表示权重和为i,有一个权重大于等于d

    f[0][0] = 1
*/

const int N = 100;
const int MOD = 1e9 + 7;

int f[N + 10][2];
int n,k,d;

int main(){
    ios::sync_with_stdio(0),cin.tie(0);
    cin >> n >> k >> d;
    f[0][0] = 1;
    for (int i = 1;i <= n; i++){
        // f[i][0] =
        for (int j = 1;j <= min(k,d-1) && i-j>=0; j++){
            f[i][0] = (f[i][0] + f[i-j][0])%MOD;
        }

        // f[i][1] =
        for (int j = 1;j <= k && i-j >= 0; j++){
            f[i][1] = (f[i][1] + f[i-j][1])%MOD;
            if (j >= d){
                f[i][1] = (f[i][1] + f[i-j][0])%MOD;
            }
        }
    }
    cout << f[n][1] << endl;
    return 0;
}