题目链接
翻译
给你一个每个节点都有 \(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;
}