k-Tree(DP)

1、问题

在这里插入图片描述

2、思路(DP)

这道题翻译过来就是说,给我们一个k叉树,然后每个点到子节点的边的边权从左到右依次为1到k。然后我们从根节点出发,向下走,我们走过的边权之和记作 n n n

现在给我们一个n,我们需要找到满足边权和为n的路径条数。同时,题目还给了我们一个d,这个d的意思就是我们的路径中至少存在一条边是大于等于d的。

我们使用DP来解决这道题:
f [ n ] f[n] f[n]表示路径和为 n n n的总条数,暂时先不管 d d d。那么这个点可以由 1 1 1 k k k表示出来。
f [ i ] = ∑ j = 1 k f [ i − j ] f[i]=\sum^k_{j=1}f[i-j] f[i]=j=1kf[ij]

那么题目中的 d d d的限制怎么体现呢?

我们可以求出不满足条件的情况,再用刚才的 f [ n ] f[n] f[n]减去即可。
对于不满足条件的情况,我们记作 g [ n ] g[n] g[n]
g [ i ] = ∑ j = 1 d − 1 g [ i − j ] g[i]=\sum^{d- 1}_{j=1}g[i-j] g[i]=j=1d1g[ij]

最终的答案就是 ( ( f [ 1 ] [ n ] − f [ 0 ] [ n ] ) ((f[1][n] - f[0][n]) % mod + mod) \% mod ((f[1][n]f[0][n])
为了方便,我们直接开一个二维数组, f [ 1 ] [ i ] f[1][i] f[1][i]是刚才的 f f f f [ 0 ] [ i ] f[0][i] f[0][i]是刚才的 g g g

3、代码

#include<bits/stdc++.h>
using namespace std;
const int mod = 1e9 + 7;
const int N = 110;
int f[2][N];
int n, k, d;

void solve()
{
	cin >> n >> k >> d;
	f[0][0] = f[1][0] = 1;
	for(int i = 1; i <= n; i ++ )
	{
		for(int j = 1; j <= k; j ++ )
			if(i >= j)
				f[1][i] = (f[1][i] + f[1][i - j]) % mod;
			
		for(int j = 1; j < d; j ++ )
		{
			if(i >= j)
				f[0][i] = (f[0][i] + f[0][i - j]) % mod;
		}
	}
	cout << ((f[1][n] - f[0][n]) % mod + mod) % mod << endl;
	
}

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	solve();
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值