codeforces 431C - k-Tree dp

题目链接：https://codeforces.ml/problemset/problem/431/C
转自：https://blog.csdn.net/qq_25955145/article/details/81261580
https://blog.csdn.net/hongrock/article/details/26557119?utm_source=blogxgwz2
题意：给一个无限的满k叉树，求从根节点开始经过的路径总长度为n，且其中至少一条边的长度不小于d的路径数。
思路：首先假设k为无穷大， 初始状态dp【1】=1
dp【2】=dp【1】+1，包括一条从路径长度为1的出来走到下一个节点p，在点p处剩余可走的长度为1，即dp【1】（dp【i】适用于每个节点），还有一条是直接走长度为2的
同理 dp【3】=dp【2】+dp【1】+1，即包括一条经过路径长度为1的，经过路径长度为2的和路径长度为3的
dp【4】=dp【3】+dp【2】+dp【1】+1
dp【n】=dp【n-1】+……+1
设dp【0】=1
则dp【n】=dp【n-1】+dp【n-2】+……dp【0】
当k不是无穷大
dp【n】=dp【n-1】+……dp【n-k】
添加d的限制
可以转换问题，求出所有边权重小于d的，再用总方案数减掉。而所有权重小于d的话，问题等价于这棵树是K值等于d-1的K-Tree，然后还是求n的方案数。
dp【i】【0】表示总方案数，dp【i】【1】表示d-1时的方案数

#include <bits/stdc++.h>
using namespace std;
long long dp[105][2];
const int mod=1e9+7;
int main()
{
    ios::sync_with_stdio(false);
    int n,k,d;
    cin>>n>>k>>d;
    dp[0][1]=dp[0][0]=1;
    d--;
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=k&&j<=i; j++)
        {
            dp[i][0]=(dp[i][0]+dp[i-j][0])%mod;
        }
        for(int j=1; j<=d&&j<=i; j++)
        {
            dp[i][1]=(dp[i][1]+dp[i-j][1])%mod;
        }
    }
    cout<<(dp[n][0]-dp[n][1]+mod)%mod<<endl;
    return 0;
}