Round 3 F - k-Tree CodeForces - 431C - K叉树 树形DP

dp[i][0]:表示权值和为 i 中不包含权值>=d的边。
dp[i][j]: 表示权值和为 i 中包含权值 >=d的边。

dp[i][0]+=dp[i-j][0] (j<d)//从 1 加到 i

dp[i][1]+=dp[i-j][0] (j>=d)

dp[i][1]+=dp[i-j][1];

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define D(v) cout<<#v<<" "<<v<<endl
#define mem(s,t) memset(s,t,sizeof(s))
const ll mod=1e9+7;
const int MAXN =105;
int n,k,d;
ll dp[2][MAXN];
int main(){
while(~scanf("%d%d%d",&n,&k,&d)){
mem(dp,0);
dp[0][0]=1;//初始化，从 0 开始设。
for(int i=0;i<=n;i++){
for(int j=1;j<=k;j++){
if(i>=j){
if(j<d){
dp[0][i]+=dp[0][i-j];
dp[0][i]%=mod;
}else {
dp[1][i]+=dp[0][i-j];
dp[1][i]%=mod;
}
dp[1][i]+=dp[1][i-j];
dp[1][i]%=mod;
}
}
}
printf("%lld\n",dp[1]
);
}
return 0;
}