ural 1056 computer net

樹上每個點都有一個距離它最遠的點，題目要求所有最遠點中，具有最小最遠距離的是哪些點

一個點，距離它最遠的點，要麼通過它的父節點到達，要麼通過它的某個孩子節點到達，顯然，一遍dfs不能得出結果

我們先做一遍dfs，求出某點通過孩子節點最多可以向下走多遠，這裏我們需要記錄最大值和次大值

然後，再做一遍dfs，求出某點通過父節點最多可以向上走多遠，如果該點父節點向下走的最遠路徑不包括該點，那麼該點向上最遠能走的距離就是父節點向下最遠能走的距離加1或者父節點向上最遠能走的距離加1，否則，爲父節點向下能走的次大值加1或者父節點向上最遠能走的距離加1，這就是爲什麼剛纔要記錄最大值和次大值

需要注意的是，次大值可以和最大值相等，判斷是否更新最大值的條件是大於等於，不是大於

代碼：

#include<iostream>
#include<memory.h>
#include<string>
#include<cstdio>
#include<algorithm>
#include<math.h>
#include<stack>
#include<queue>
#include<vector>
#include<map>
using namespace std;
const int MAX=10005;
struct node 
{
	int v,next;
}g[MAX*5];
int adj[MAX],down[MAX][2],up[MAX],res[MAX],e,n;
void add(int u,int v)
{
	g[e].v=v; g[e].next=adj[u]; adj[u]=e++;
}
void dfs1(int u,int fa,int d)
{
	int i,v;
	down[u][0]=down[u][1]=0;
	for(i=adj[u];i!=-1;i=g[i].next)
	{
		v=g[i].v;
		dfs1(v,u,d+1);
		if(down[u][0]<=down[v][0]+1)
		{
			down[u][1]=down[u][0];
			down[u][0]=down[v][0]+1;
		}
		else if(down[u][1]<=down[v][0]+1)
			down[u][1]=down[v][0]+1;
	}
}
void dfs2(int u,int fa)
{
	int i,v;
	if(fa!=-1)
	{
		if(down[u][0]==down[fa][0]-1)
			up[u]=max(down[fa][1],up[fa])+1;
		else
			up[u]=max(down[fa][0],up[fa])+1;
	}
	res[u]=max(up[u],down[u][0]);	
	for(i=adj[u];i!=-1;i=g[i].next)
	{
		v=g[i].v;
		dfs2(v,u);
	}
}		
int main()
{
	int i,j,k;
	while(scanf("%d",&n)!=EOF)
	{
		memset(adj,-1,sizeof(adj));
		e=0;
		for(i=2;i<=n;i++)
		{
			scanf("%d",&j);
			add(j,i);
		}
		dfs1(1,-1,0);
		dfs2(1,-1);
		int Min=99999999;
		for(i=1;i<=n;i++)
			Min=min(Min,res[i]);
		for(i=1;i<=n;i++)
		{
			if(res[i]==Min)
				printf("%d ",i);
		}
		printf("\n");
	}
	return 0;
}