k-Tree CodeForces - 431C

原题链接

考察:树形dp

思路:

        设f[i][j]表示和为i,j==1表示经过了>=d的边,j=0表示未经过的方案数.以最后一步来划分集合,最后一步的和为i-j,此时边为j,那么状态转移方程是:

        if(j>=d) f[i][1] = f[i-j][0]+f[i-j][1]

        else f[i][1] = f[i-j][1] ,f[i][0] = f[i-j][0]

        初始化f[0][0] = 1,也就是根节点的方案数为1.

注意:最后算出答案也不要忘了取模！！！

这道题用记忆化搜索没做出来,我果然菜

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 110,Mod = 1000000007;
int n,k,d;
int f[N][2];
int main() 
{
    scanf("%d%d%d",&n,&k,&d);
    f[0][0] = 1;
    for(int i=1;i<=n;i++)
      for(int j=1;j<=k;j++)
      {
          if(j>i) continue;
          if(j>=d) f[i][1] = (f[i-j][0]+f[i-j][1])%Mod+f[i][1]%Mod;
          else f[i][1] = (f[i-j][1]+f[i][1])%Mod,f[i][0] = (f[i][0]+f[i-j][0])%Mod;
          f[i][1]%=Mod;
          f[i][0]%=Mod;
      }
    printf("%d\n",f[n][1]);
    return 0;
}