ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

YbtOJ-森林之和【dp】

2022-02-07 21:03:24  阅读:154  来源: 互联网

标签:子树 YbtOJ ll 权值 指定 ans 节点 dp 森林


正题


题目大意

一个节点的权值定义为它度数的平方,求所有\(n\)个点的有标号森林的所有节点权值和。

\(1\leq n,T\leq 5\times 10^3\)


解题思路

首先因为所有节点本质相同,所以我们可以只考虑一个节点所有情况下的权值和。

然后考虑这个平方和怎么做,我们可以视为指定一个节点连出两颗子树的方案(可以相同)。

那么考虑这个怎么做,首先我们需要处理出\(n\)个节点有根树和无根树的数组\(r,f\)。

然后我们要考虑怎么统计除了指定子树以外的方案,首先我们需要处理出\(n\)个点的森林个数\(s_n\)。

我们可以考虑每次枚举新加入的树的大小,但是要指定这个节点编号最小的节点编号必须是\(1\)(以防相同的子树算重),那么有

\[s_n=\sum_{i=1}^ns_{n-i}f_{i}\binom{n-1}{i-1} \]

然后还要算上非指定的子树中和\(1\)号点联通的其他节点的方案,那么有

\[g_n=\sum_{i=0}^ns_ir_{i+1}\binom{n}{i} \]

至于指定子树的话,我们枚举指定子树的大小转移就好了。

时间复杂度:\(O(n^2+T)\)


code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=5100;
ll T,P,C[N][N],f[N],r[N],g[N],s[N],d[N],ans[N];
ll power(ll x,ll b){
	ll ans=1;
	while(b){
		if(b&1)ans=ans*x%P;
		x=x*x%P;b>>=1;
	}
	return ans;
}
signed main()
{
	freopen("forest.in","r",stdin);
	freopen("forest.out","w",stdout);
	scanf("%lld%lld",&T,&P);
	C[0][0]=1;
	for(ll i=1;i<N;i++)
		for(ll j=0;j<=i;j++)
			C[i][j]=(C[i-1][j]+(j?C[i-1][j-1]:0))%P;
	f[0]=f[1]=r[0]=r[1]=g[0]=s[0]=1;
	for(ll i=2;i<N;i++) 
		f[i]=power(i,i-2),r[i]=f[i]*i%P;
	for(ll i=1;i<N;i++)
		for(ll j=0;j<i;j++)
			(s[i]+=s[j]*f[i-j]%P*C[i-1][j]%P)%=P;
	for(ll i=1;i<N;i++){
		for(ll j=0;j<=i;j++)
			(g[i]+=s[j]*f[i-j+1]%P*C[i][j]%P)%=P;
		for(ll j=1;j<i;j++)
			(ans[i]+=r[j]*g[i-j-1]%P*C[i-1][j]%P)%=P;
	}
	for(ll i=1;i<N;i++){
		d[i]=ans[i+1];
//		for(ll j=1;j<=i;j++)
//			(d[i]+=r[j]*g[i-j]%P*C[i][j]%P)%=P;
		for(ll j=1;j<i-1;j++)
			(ans[i]+=d[j]*r[i-j-1]%P*C[i-1][j]%P)%=P;
	}
	while(T--){
		ll n;scanf("%lld",&n);
		printf("%lld\n",ans[n]*n%P);
	}
	return 0;
}

标签:子树,YbtOJ,ll,权值,指定,ans,节点,dp,森林
来源: https://www.cnblogs.com/QuantAsk/p/15869160.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有