码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ1131 [POI2008]Sta 其他

时间:2017-12-21 17:15:52      阅读:136      评论:0      收藏:0      [点我收藏+]

标签:blog   algorithm   i++   clu   amp   span   题解   http   size   

原文链接http://www.cnblogs.com/zhouzhendong/p/8081100.html


题目传送门 - BZOJ1131


题意概括

  给出一个N个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大。


题解

  嘻,这题不卡栈。

  假设以1为根

  先跑一遍dfs,算出每一个子树的节点数size,同时算出以1为根节点的深度和。

  然后再跑一遍dfs,这一回,我们就可以算答案了。

  假设我们要把树根从一条边的一个节点移向另一个节点,那么,这两个节点为根的答案差就是这条边两端的节点个数差。因为其中一个节点代表的子树上的节点都要多走一步到根,而另一边少走一步。这样就可以在O(n)的时间复杂度内solve这一题了。


代码

#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long LL;
const int N=1000005;
struct Gragh{
	int cnt,y[N*2],nxt[N*2],fst[N];
	void clear(){
		cnt=0;
		memset(fst,0,sizeof fst);
	}
	void add(int a,int b){
		y[++cnt]=b,nxt[cnt]=fst[a],fst[a]=cnt;
	}
}g;
int n,size[N],ans;
LL sum[N],res[N];
void dfs1(int rt,int pre){
	size[rt]=1,sum[rt]=0;
	for (int i=g.fst[rt];i;i=g.nxt[i])
		if (g.y[i]!=pre){
			int s=g.y[i];
			dfs1(s,rt);
			size[rt]+=size[s];
			sum[rt]+=sum[s];
		}
	sum[rt]+=size[rt]-1;
}
void dfs2(int rt,int pre){
	res[rt]=res[pre]-size[rt]+(n-size[rt]);
	if (!ans||res[rt]>res[ans]||(res[rt]==res[ans]&&rt<ans))
		ans=rt;
	for (int i=g.fst[rt];i;i=g.nxt[i])
		if (g.y[i]!=pre)
			dfs2(g.y[i],rt);
}
int main(){
	g.clear();
	scanf("%d",&n);
	for (int i=1,a,b;i<n;i++){
		scanf("%d%d",&a,&b);
		g.add(a,b);
		g.add(b,a);
	}
	dfs1(1,0);
	ans=0;
	res[0]=sum[1]+n;
	dfs2(1,0);
	printf("%d",ans);
	return 0;
}

  

BZOJ1131 [POI2008]Sta 其他

标签:blog   algorithm   i++   clu   amp   span   题解   http   size   

原文地址:http://www.cnblogs.com/zhouzhendong/p/8081100.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!