*1600第二题,慢慢做,慢慢想。
题目链接:点击打开原题目
题意:给一个数字k,用它构造一棵树,构造方法是,从根节点开始往下,每个节点都有k个子节点(所以这棵树的深度是无限的,一开始没有读到这一点一直想不通样例),每条连接子节点的边的权值从1~k,现在要求从根节点出发的所有路径中,路径权值总和为n,且路径上至少有一条边的权值大于等于d的方案有多少种。
思路:用 dp[i][j] 表示当前走的路径的总和为 i,路径中的最大的权值为 j 的方案数。那么转移方程是dp[i][j]+=dp[i-c][j],表示当前总和为i,最大值为j的情况,可以由上一步总和为i-c最大值还是j的情况,走一条权值为c的边这种情况得到,将所有可能的情况加起来就是dp[i][j]的方案数。而我们最后要得到的是,最大为k的数组成的,但是最小要有一个大小为d的边,利用容斥定理可以知道,只需要用所有的方案数减去不合法的,也就是最大只有d-1的数组成的方案数就是我们最后要得到的答案(开long long 啊)。
AC代码:
#include<bits/stdc++.h>
using namespace std;
const int mod=1000000007;
long long dp[110][110];//dp[i][j]表示已经选了i个数,最大的一个数是j
int main()
{
int n,k,d;
scanf("%d%d%d",&n,&k,&d);
//用小于k的数凑成n。
for(int i=1; i<=k; i++)
dp[0][i]=1;
for(int i=1; i<=n; i++) //枚举最后要得到的和
for(int j=1; j<=k; j++) //枚举当前路径中最大的值
for(int c=1; c<=j; c++) //枚举当前能选的数,最大是k
{
if(i-c>=0)
dp[i][j]=(dp[i][j]+dp[i-c][j])%mod;
else break;
}
printf("%lld\n",(dp[n][k]-dp[n][d-1]+mod)%mod);
return 0;
}