k-Tree CodeForces - 431C

http://codeforces.com/problemset/problem/431/C

dp[i][j]代表走到第(i+1)层权值和为n的方案数 转移方程为dp[i][k+j]+=dp[i-1][k] 其中j是当前边的权值

题目要求包含至少一条权值大于等于d的边 用总方案数减去一条都不包含的即可

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1000000007;
const int maxn=1e2+10;

ll dp1[maxn][maxn],dp2[maxn][maxn];
int n,kk,dd;

void solve(ll dp[][maxn],int lim)
{
    int i,j,k;
    for(j=1;j<=lim;j++){
        dp[1][j]=1ll;
    }
    for(i=2;i<=n;i++){
        for(j=1;j<=lim;j++){
            for(k=1;k+j<=n;k++){
                dp[i][k+j]=(dp[i][k+j]+dp[i-1][k])%mod;
            }
        }
    }
}

int main()
{
    ll ans;
    int i;
    scanf("%d%d%d",&n,&kk,&dd);
    solve(dp1,kk);
    solve(dp2,dd-1);
    ans=0;
    for(i=1;i<=n;i++){
        ans=(ans+dp1[i][n]-dp2[i][n]+mod)%mod;
    }
    printf("%lld\n",ans);
    return 0;
}