Java教程

TediousLee(DP+思维)

本文主要是介绍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;
} 
这篇关于TediousLee(DP+思维)的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!