Codeforces431C_K-Tree_KEY

Posted 2020-10-10 Cptraser

题目传送门

题目大意：给定一棵K叉树，一个节点下的K个节点的分数为i，求包含大于等于D的边的总分数为N的方案总数，mod 1e9+7。

这是一道很好的树形DP题，首先看N和K两个条件，可以轻易的得到方程：

f[i][j]=(f[i][j]+f[i-k][j])%mod;

//f[i][j]表示总分为i,当前边分数到j的方案总数。

再看最后一个条件，我们仔细一想，不是只要将答案f[N][K]减去f[N][D-1]即可吗？

只需要减去小于D的所有方案不就好了吗？

answer=f[N][K]-f[N][D-1]；

。。。。。。

所以这道题还是挺水的。

code：

 

#include <cstdio>
#define mod 1000000007 
using namespace std;
int N,K,D;
int f[101][101];
int main(){
    scanf("%d%d%d",&N,&K,&D);
        for(int i=1;i<=K;i++)f[0][i]=1;
        for(int i=1;i<=N;i++)
            for(int j=1;j<=K;j++)
                for(int k=1;k<=j;k++){
                    if(i-k>=0)f[i][j]=(f[i][j]+f[i-k][j])%mod;
                    else break;
                }
    printf("%d",(f[N][K]-f[N][D-1]+mod)%mod);
    return 0;
}