AtCoder Grand Contest 018D Tree and Hamilton Path 树的重心 贪心 乱搞

Description

---

有一棵n个节点带边权的树，建一个完全图G，G中两点的边权为树上两点的最短路，在G中找一条权值之和最大的哈密顿回路，求这个权值
$n\le 10^5$

Solution

---

实际上就是要找一个排列p[]，使得$\sum dis(p_i,p_{i-1})$最小
贪心地想，我们保证每次经过树根一定最优，因为这样两两lca都在树根处。
那么问题变成定一个根使得每个节点都能在所在子树外找到匹配点
可以发现这恰好就是重心的情况，由于每个子树大小不超过n/2，因此是可以一一匹配完的

两个重心的情况要讨论一下

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const int N=100005;

std:: priority_queue <LL> heap;

struct edge {int y,w,next;} e[N*2];

int mx[N],size[N],rt,rt2;
int ls[N],edCnt,n;

LL dis[N];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void add_edge(int x,int y,int w) {
	e[++edCnt]=(edge) {y,w,ls[x]}; ls[x]=edCnt;
	e[++edCnt]=(edge) {x,w,ls[y]}; ls[y]=edCnt;
}

void dfs(int x,int fa) {
	size[x]=1;
	for (int i=ls[x];i;i=e[i].next) {
		if (e[i].y==fa) continue;
		dis[e[i].y]=dis[x]+e[i].w;
		dfs(e[i].y,x); size[x]+=size[e[i].y];
		mx[x]=std:: max(mx[x],size[e[i].y]);
	}
	mx[x]=std:: max(mx[x],n-size[x]);
	if (mx[x]<mx[rt]) rt=x;
}

int main(void) {
	n=read();
	rep(i,2,n) {
		int x=read(),y=read(),w=read();
		add_edge(x,y,w);
	}
	mx[rt]=n;
	dfs(1,0);
	dis[rt]=0;
	dfs(rt,0);
	LL ans=0,sum=0; bool flag=false;
	rep(i,1,n) sum+=dis[i]*2;
	for (int i=ls[rt];i;i=e[i].next) {
		if (mx[e[i].y]==mx[rt]) {
			ans=sum-e[i].w;
			flag=true;
		}
	}
	if (!flag) rep(i,1,n) if (i!=rt) {
		ans=std:: max(ans,sum-dis[i]);
	}
	printf("%lld\n", ans);
	return 0;
}