codeforces 431C k-Tree (树形dp)

题目链接

题意

• 给定一颗 K树，K 树每个结点以下的分支都为 K 条（每个点都有 K 个子节点），边权分别为 1 — K。
• 求从根节点开始，总长度为 n ，同时路径上至少有一条边的长度超过 d 的路径有多少条。

k=3时,图如下 

思路 

1. 树的大小未知,但是当每条线的权值为1时,我们最多能取到n层,所以树的大小最大为n
2. 树形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;
}