F - k-Tree (dp,递归)

给定一个k叉树，且是无限有根树，每个节点有k个孩子，从左向右依次编号为1,2,3,...k，每条边有权值，节点连向i号孩子的边的权值为i，问：从根节点开始向下走，有多少条路径满足要求：该路径权值为ｎ，且路径上至少有一条边的权值大于等于ｄ

 

 

dp[i]:权值为i的路径的条数。
dp2[i]:权值为i的，且路径上的权值都小于d的路径条数。

利用了递归的思想，当dp[i-j] （i-j>=0)时 就回到了求i-j个节点的树的问题

所以 dp[i]+=dp[i-j]

 

#include <iostream>
#include<algorithm>
#include<vector>
#include<cstring>
#define N 105
using namespace std;

int dp[105],dp1[105];
int n,k,d;

int main()
{
    cin>>n>>k>>d;
    dp[0]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=k;j++)
        {
            if(i-j>=0)
            {
                dp[i]+=dp[i-j];
                dp[i]%=1000000007;
            }
        }
    }
    dp1[0]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<d;j++)
        {
            if(i-j>=0)
            {
                dp1[i]+=dp1[i-j];
                dp1[i]%=1000000007;
            }
        }
    }
    long long sum=(dp[n]-dp1[n])%1000000007;
    if(sum<0) sum+=1000000007;
    cout<<sum<<endl;
}