POJ 2631 树的直径

树的直径

问题：树的直径定义为一棵树上两个节点间的路径长度的最大值。
分析：
树形 dp：我们以任意一个点为根，向下进行遍历。经过 u 点，而不经过其父节点的最长路
径的两个端点，一定是在他的不同的子树中的两个叶子节点，因此我们可以记录 u 点的各个子
树的深度的最大值和次大值。累加即为经过这个节点时的最长路经长度。当然最长路径也可能从
父节点经过，那就是父节点的问题了。最后输出所有点最大值和次大值之和的最大。
假设 s-t这条路径为树的直径，或者称为树上的最长路
现有结论，从任意一点u出发搜到的最远的点一定是 s、t中的一点，然后再从这个最远点
开始搜，就可以搜到另一个最长路的端点，即用两遍广搜就可以找出树的最长路。
习题：POJ 2631

【AC代码】：

#include<cstdio>
#include<vector>
#include<cstring>
#define M(a,b) memset(a,b,sizeof(a))
using namespace std;

inline void read(int &x){
    char ch=getchar(),c=ch;
	x=0;
    while(ch<'0' || ch>'9'){
    	 c=ch;
		 ch=getchar();
	}
    while(ch>='0' && ch<='9'){
    	x=(x<<1)+(x<<3)+ch-'0';
		ch=getchar();
	}
    if(c=='-')x=-x;
}

const int maxn=10005;
vector<pair<int,int> > G[maxn];
int dis[3][maxn];
bool vis[maxn];
int d,p;

void DFS(int u,int flag){
	vis[u]=true;
	if(dis[flag][u]>d){
		d=dis[flag][u];
		p=u;
	}
	for(int i=0;i<G[u].size();i++){
		int v=G[u][i].first;
		int w=G[u][i].second;
		if(!vis[v]){
			dis[flag][v]=dis[flag][u]+w;
			DFS(v,flag);
		}
	}
}

int main(){
	int u,v,w;
    while(scanf("%d %d %d",&u,&v,&w)!=EOF){
        G[u].push_back(make_pair(v,w));
        G[v].push_back(make_pair(u,w));
    }
    memset(dis,0,sizeof(dis));
    memset(vis,false,sizeof(vis));
    d=0;
    DFS(1,0);
    memset(vis,false,sizeof(vis));
    d=0;
    DFS(p,1);
    printf("%d\n",d);
    return 0;
}