pku 1655 Balancing Act (树状)

题目：对于一棵树，删除其中一个节点后，可以得到若干子树，这些子树的规模的最大值叫balance。给定一棵树，问拆除哪个节点，可以得到最小的balance。

解法：预处理，对给定的树，任意选择一个节点作为根节点（我选择的1号节点），DFS整棵树，得到每个节点的子节点个数。枚举删除每个节点的情况，即可得到结果。

#include<iostream>
#include<algorithm>
using namespace std;
const int maxN=20005;
struct Edge
{
	int s,t;
}edges[2*maxN];
bool operator<(const Edge &a, const Edge &b)
{
	if(a.s<b.s) return true;
	else if (a.s==b.s&&a.t<b.t) return true;
	else return false;
}
int edgeIndex[maxN];
struct Node
{
	int father;
	int cnt;
}nodes[maxN];
bool visited[maxN];
int PreProcess(int s,int father)
{
	visited[s]=true;
	int cnt=1;
	for(int i=edgeIndex[s];i<edgeIndex[s+1];i++)
	{
		if(visited[edges[i].t]==true) continue;
		cnt+=PreProcess(edges[i].t,s);
	}
	nodes[s].father=father;
	nodes[s].cnt=cnt;
	return cnt;
}
int DeleteThis(int s,int N)
{
	int balance=0,sonNode=0;
	for(int i=edgeIndex[s];i<edgeIndex[s+1];i++)
	{
		if(nodes[edges[i].t].father==s)
		{
			balance=max(balance,nodes[edges[i].t].cnt);
			sonNode+=nodes[edges[i].t].cnt;
		}
	}
	balance=max(balance,N-sonNode-1);
	return balance;
}
int main()
{
	//freopen("f:\\in.txt","r",stdin);
	int n;
	scanf("%d",&n);
	while(n--)
	{
		int N,s,t=0;
		scanf("%d",&N);
		int edgeNum=N-1;
		for(int i=0;i<edgeNum;i++)
		{
			scanf("%d%d",&s,&t);
			edges[i+edgeNum].t=edges[i].s=s;
			edges[i+edgeNum].s=edges[i].t=t;
		}
		sort(edges,edges+2*edgeNum);
		edgeIndex[1]=0;
		for(int i=2;i<=N;i++)
		{
			edgeIndex[i]=edgeIndex[i-1];
			while(edges[edgeIndex[i]].s==i-1) edgeIndex[i]++;
		}
		edgeIndex[N+1]=edgeNum*2;

		memset(visited,0,sizeof(visited));
		PreProcess(1,0);

		int ans,balance=maxN;
		for(int i=1;i<=N;i++)
		{
			int tmp=DeleteThis(i,N);
			if(tmp<balance)
			{
				balance=tmp;
				ans=i;
			}
		}
		printf("%d %d\n",ans,balance);
	}
}