[bzoj1131][POI2008]Sta_树形dp

最新推荐文章于 2018-12-17 20:07:20 发布

转载最新推荐文章于 2018-12-17 20:07:20 发布 · 43 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/ShuraK/p/9398844.html

本文介绍了一个算法问题，旨在寻找一棵给定树中使深度和达到最大的根节点。通过对树进行深度优先搜索和节点价值重估，最终确定最优根节点。代码实现采用C++，展示了树结构的构建、深度计算及节点价值更新过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Sta bzoj-1131 POI-2008

题目大意：给定一棵n个点的树，求一个根，使得深度和最大。

注释：$1\le n \le 10^6$。

想法：扭一扭即可。

扭的时候看看这个点当没当过根。

最后，附上丑陋的代码... ...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 1000010 
using namespace std;
typedef long long ll;
int to[N<<1],nxt[N<<1],head[N],tot,size[N],dis[N],n;
ll val[N],ans=0;
int now;
inline char nc()
{
	static char buf[100000],*p1,*p2;
	return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
int read()
{
	int x=0; char c=nc();
	while(!isdigit(c)) c=nc();
	while(isdigit(c)) x=(x<<1)+(x<<3)+c-'0',c=nc();
	return x;
}
inline void add(int x,int y)
{
	to[++tot]=y;
	nxt[tot]=head[x];
	head[x]=tot;
}
void dfs(int pos,int fa)
{
	size[pos]=1,dis[pos]=dis[fa]+1;
	for(int i=head[pos];i;i=nxt[i])
	{
		if(to[i]==fa) continue;
		dfs(to[i],pos);
		size[pos]+=size[to[i]];
	}
}
void dispose(int pos,int fa)
{
	val[pos]=val[fa]-size[pos]+(n-size[pos]);
	for(int i=head[pos];i;i=nxt[i])
	{
		if(to[i]==fa) continue;
		dispose(to[i],pos);
	}
}
int main()
{
	n=read();
	for(int x,y,i=1;i<n;i++)
	{
		x=read(); y=read();
		add(x,y); add(y,x);
	}
	dfs(1,1);
	for(int i=1;i<=n;i++) val[1]+=dis[i]/* ,printf("%d\n",dis[i]) */;
	dispose(1,1);
	for(int i=1;i<=n;i++) if(val[i]>ans) ans=val[i],now=i;
	printf("%d\n",now);
	return 0;
}

小结：有意思... ...

转载于:https://www.cnblogs.com/ShuraK/p/9398844.html