Round 3 F

来源：互联网发布：如何取淘宝店名编辑：程序博客网时间：2024/04/30 03:42

题目链接：
http://codeforces.com/problemset/problem/431/C

大意：
无限的完全 K 叉树，每层边权值分别为 1.2.3…K
要求至少经过一个大于 d 的边
求和为 n 有多少种走法

思路：
树形DP，一道很好的 dp 题，方程推出来了，但是想歪了，一开始甚至想三维dp[i][j][k] 准备用上 n k d
实际上按照递推方程的写法来思考考虑 k 的时候，就不要考虑 d 的变化
解法是把 d 和 k 叉树看成一个状态

dp[i][0]:表示权值和为 i 中不包含权值>=d的边。dp[i][j]: 表示权值和为 i 中包含权值 >=d的边。dp[i][0]+=dp[i-j][0] (j<d)//从 1 加到 idp[i][1]+=dp[i-j][0] (j>=d)dp[i][1]+=dp[i-j][1];

代码：

#include<bits/stdc++.h>using namespace std;typedef long long ll;#define D(v) cout<<#v<<" "<<v<<endl#define mem(s,t) memset(s,t,sizeof(s))const ll mod=1e9+7;const int MAXN =105;int n,k,d;ll dp[2][MAXN];int main(){    while(~scanf("%d%d%d",&n,&k,&d)){        mem(dp,0);        dp[0][0]=1;//初始化，从 0 开始设。        for(int i=0;i<=n;i++){            for(int j=1;j<=k;j++){                if(i>=j){                    if(j<d){                        dp[0][i]+=dp[0][i-j];                        dp[0][i]%=mod;                    }else {                        dp[1][i]+=dp[0][i-j];                        dp[1][i]%=mod;                    }                    dp[1][i]+=dp[1][i-j];                    dp[1][i]%=mod;                }            }        }        printf("%lld\n",dp[1][n]);    }    return 0;}

参考博客：
http://blog.csdn.net/u014634338/article/details/45722203
http://blog.csdn.net/hongrock/article/details/26557119

阅读全文

0 0