【CF431C】C. k-Tree（树形dp）

K树

题意：

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

思路：

• 很显然的树形dp…但没给定树的大小，要自行寻找范围。
• 显然我们最多需要处理 n 层即可，n 层每层长度为 1 的路径满足条件，再深的层会超出 n 长度限制。
• 同时因为 K 树的独特性质，可以在转移的时候用任何长度的边转移。

$Code:$

#include<bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define forr(a,b,c) for(int a=b;a<=c;a++)
#define rfor(a,b,c) for(int a=b;a>=c;a--)
#define all(a) a.begin(),a.end()
#define oper(a) (operator<(const a& ee)const)
#define endl "\n"
using namespace std;

typedef long long ll;
typedef pair<int, int> PII;

double DNF = 1e17;
const int N = 110, M = 400010, MM = 110;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int INF = 0x3f3f3f3f, mod = 1e9 + 7;
int n, m, k, T, S, D, K;
ll dp[N][2];

void solve() {
	cin >> n >> k >> D;
	dp[0][0] = 1;//边界

	for (int i = 1; i <= n; i++) { //长度为 i 的路径
		for (int j = 1; j <= k; j++) { 
			// j 枚举存在的边
			if (i - j < 0)continue;//合法状态
			//任意长度为 i 的路径都可以由 i - j 转移过来

			if (j < D) { //D的限制
				dp[i][1] += dp[i - j][1];
				dp[i][0] += dp[i - j][0];
				dp[i][1] %= mod;
				dp[i][0] %= mod;
			}
			else {
				dp[i][1] += dp[i - j][1] + dp[i - j][0];
				dp[i][1] %= mod;
			}
		}
	}
	cout << dp[n][1];
}

int main() {
	cinios;
	T = 1;
	while (T--)solve();
	return 0;
}
/*
*/