TediousLee(DP+思维)

题目链接:Problem - 1369D - Codeforces TediousLee

题意不难理解,这棵树的每一阶都是由之前一阶中所有叶子节点下加一个叶子,所有有一个子树的节点加两个叶子所形成,我们需要求 n n n​​​阶树中有多少个”爪型“,然后结果乘以4就行,数据很大需要进行取模。不难想到dp, d p [ i ] [ 3 ] dp[i][3] dp[i][3]​​​,其中 d p [ i ] [ 0 ] dp[i][0] dp[i][0]​​​表示没有子树的点 ,其中 d p [ i ] [ 1 ] dp[i][1] dp[i][1]​​​表示只有一个子树的点,其中 d p [ i ] [ 2 ] dp[i][2] dp[i][2]​​​​​​​​表示有三个子树的点。状态转移方程就比较容易想了(见代码注释)搞清楚转移后就需要考虑如何计数了。参考链接中的原题的图,这是一颗4阶的树,它的答案是,显然他只能在(1,2,3,4)和(2,5,6,7)中选择其中一个组合。其中2是这一轮新增的有三个子树的节点,这个值必然是会越来越大的,这个值必须要记在答案中,这个值如果呗记在答案中,那么前两次新加的这个值就不能加入到答案中,因为他们会重复如果不能理解,建议自己画个图。

#include<bits/stdc++.h>
#define ll long long
const int mod=1e9+7;
const int N=2e6+10;
using namespace std;
int t,n;
ll f[N][3],ans[N];
int main(){
	f[1][0]=1;
	//cout<<"##"<<f[1][0]<<" "<<f[1][1]<<" "<<f[1][2]<<endl;	
	for(int i=2;i<=2e6;i++){
		f[i][0]=(f[i-1][0]+f[i-1][1]*2)%mod;//叶子节点可以由没有子树的点产生1个,														//由有一个子树的点产生2个
		f[i][2]=(f[i-1][2]+f[i-1][1])%mod;//有三个子树的点会增加前一阶段有一个子树的点
		f[i][1]=(f[i-1][0])%mod;//有一个子树的点可以由叶子节点得到
		//cout<<"##"<<f[i][0]<<" "<<f[i][1]<<" "<<f[i][2]<<endl;
		ans[i]=(f[i][2]-f[i-1][2]+mod)%mod*4%mod;
		if(i>=3) ans[i]=(ans[i]+ans[i-3])%mod;	
	}
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		printf("%lld\n",ans[n]);
	}
	return 0;
} 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值