Codeforces 431C - k-Tree

431C - k-Tree

思路：dp.

dp[i][j][s]

如果s为1，表示第i层长度为j且至少包含一段>=d的距离的路径数

如果s为0，表示第i层长度为j且不包含一段>=d的距离的路径数

状态转移看代码

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int MOD=1e9+;
const int N=;
ll dp[N][N][];
int main(){
    ios::sync_with_stdio(false);
    cin.tie();
    int n,k,d;
    cin>>n>>k>>d;
    for(int i=;i<=k;i++)if(i>=d)dp[][i][]=;else dp[][i][]=;
    for(int i=;i<=n;i++){
        for(int j=;j<=n;j++){
            for(int ii=;ii<=min(j,k);ii++){
                if(ii>=d)dp[i][j][]=(dp[i][j][]+dp[i-][j-ii][]+dp[i-][j-ii][])%MOD;
                else dp[i][j][]=(dp[i][j][]+dp[i-][j-ii][])%MOD,dp[i][j][]=(dp[i][j][]+dp[i-][j-ii][])%MOD;
            }
            //cout<<dp[i][j][1]<<' ';
        }
        //cout<<endl;
    }
    ll ans=;
    for(int i=;i<=n;i++)ans=(ans+dp[i][n][])%MOD;
    cout<<ans<<endl;
    return ;
}