题意
- 给定一颗 K树,K 树每个结点以下的分支都为 K 条(每个点都有 K 个子节点),边权分别为 1 — K。
- 求从根节点开始,总长度为 n ,同时路径上至少有一条边的长度超过 d 的路径有多少条。
k=3时,图如下
思路
- 树的大小未知,但是当每条线的权值为1时,我们最多能取到n层,所以树的大小最大为n
- 树形dp,dp[i][0]代表总权重为i时且没有经过>=d的边的方案数,dp[i][1]代表总权重为i时且有经过>=d的边的方案数
code
#include<bits/stdc++.h>
#define PII pair<int,int>
#define ls u<<1
#define rs u<<1|1
#define xx first
#define yy second
#define lowbit(i) i&-i
#define int long long
using namespace std;
const int N = 5e5+2;
const int mod = 1e9+7;
int dp[110][2];
void sol()
{
int n,k,d;
cin>>n>>k>>d;
dp[0][0]=1;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=k;j++)
{
if(i-j<0) continue;
if(j<d)
{
dp[i][1] = (dp[i][1] + dp[i-j][1])%mod;
dp[i][0] = (dp[i][0] + dp[i-j][0])%mod;
}
else dp[i][1] = (dp[i][1] + dp[i-j][0] + dp[i-j][1])%mod;
}
}
cout<<dp[n][1]<<endl;
}
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
int t;
t=1;
// cin>>t;
while(t--)
sol();
return 0;
}