BZOJ2526: [Poi2011]Inspection

题目大意：给定一颗n个节点的树，要求求出从每个节点开始的遍历最小代价。一次遍历是指，每次从这个点出发去一个点再回来，代价是这条边边权*2，相邻两次选择的点对于这个点来说不能在同一颗子树里，最后一次可以不用回来，总代价就是每次出去再回来的代价之和

刚一看题，这不SB题吗，一个treeDP维护出来最长链次长链不就完事了吗，每次从一个点转移过来的时候就各种讨论一下就好了

判断误无解的时候，只需要看一下对于这个点最大的那个子树大小超没超过一半，算答案就是这个点对于所有点出去再回来的权值和减去最长的那个

然后就写了一发，WA了，然后怎么调都是WA

后来看了数据，发现想少了一种情况，就是对于一个点假如正好有一个子树大小为n/2，那就必须第一次走这个子树的，最后一次也走这个子树的，也就是说不能减去最长链，必须减去在这个子树里的最长链.....

太坑了....

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000010
using namespace std;
int to[N<<1],nxt[N<<1],pre[N],cnt;
void ae(int ff,int tt)
{
	cnt++;
	to[cnt]=tt;
	nxt[cnt]=pre[ff];
	pre[ff]=cnt;
}
int fa[N];
long long tot[N];
int fir[N],sec[N],mx[N],siz[N];
void build(int x)
{
	int i,j;
	siz[x]=1;
	for(i=pre[x];i;i=nxt[i])
	{
		j=to[i];
		if(j==fa[x]) continue;
		fa[j]=x;
		build(j);
		tot[x]+=tot[j]+siz[j];
		siz[x]+=siz[j];
		mx[x]=max(mx[x],siz[j]);
		if(fir[j]+1>sec[x]) sec[x]=fir[j]+1;
		if(fir[x]<sec[x]) swap(fir[x],sec[x]);
	}
}
int n;
long long ans[N];
void dfs(int x,int lg,long long F)
{
	int i,j;
	if(max(mx[x],n-siz[x])>n/2) ans[x]=-1;
	else if(max(mx[x],n-siz[x])==n/2&&n%2==0)
	{
		if(n-siz[x]==n/2) ans[x]=2*tot[x]+F-lg;
		else
		{
			for(i=pre[x];i;i=nxt[i])
			{
				j=to[i];
				if(siz[j]==mx[x]) ans[x]=2*tot[x]+F-fir[j]-1;
			}
		}
	}
	else ans[x]=2*tot[x]+F-max(fir[x],lg);
	for(i=pre[x];i;i=nxt[i])
	{
		j=to[i];
		if(j==fa[x]) continue;
		if(fir[x]==fir[j]+1)
		dfs(j,max(sec[x],lg)+1,F+2*(tot[x]-tot[j]+n-siz[j]-siz[j]));
		else
		dfs(j,max(fir[x],lg)+1,F+2*(tot[x]-tot[j]+n-siz[j]-siz[j]));
	}
}
int main()
{
	int x,y,i,j;
	scanf("%d",&n);
	for(i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		ae(x,y);ae(y,x);
	}
	build(1);
	dfs(1,0,0);
	for(i=1;i<=n;i++)
	printf("%lld\n",ans[i]);
}