HDU 2196 树的直径

到树上的任意节点最远的点一定是树的两个端点之一

要注意bfs判断端点的时候不能再用记最后出队元素的方法 

因为边权是给定的 不一定为1

#include<algorithm>
#include<stdio.h>
#include<iostream>
#include<vector>
#include<queue>
#define maxn 10005
using namespace std;
#define inf 1000000005
vector<int>G[maxn];
int vis[maxn];
int d1[maxn],d2[maxn];
int s,n;
struct Edge
{
    int from,to,va;
};
vector<Edge>edges;
void AddEdge(int from,int to,int va)
{
    edges.push_back((Edge){from,to,va});
    edges.push_back((Edge){to,from,va});
    int k =(int)edges.size();
    G[from].push_back(k-2);
    G[to].push_back(k-1);
}
void bfs(int *d,int &s)
{
    for(int i=1;i<=n;++i)
    {
        d[i]=-1;
        vis[i]=0;
    }
    queue<int>q;
    q.push(s);
    vis[s]=1;
    d[s]=0;
    
    while(!q.empty())
    {
          int k=q.front();
           q.pop();
        if(d[s]<d[k])s=k;     //因为边权不一定为1 所以最后出队的元素不一定是端点
        
        for(int i=0;i<G[k].size();++i)
        {
            Edge &x=edges[G[k][i]];
         
            if(vis[x.to])continue;
               d[x.to]=d[k]+x.va;
                      vis[x.to]=1;
                q.push(x.to);
        }
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        memset(d1,0,sizeof d1);
        memset(d2,0,sizeof d2);
        int a,va;
        for(int i=2;i<=n;++i)
        {
            scanf("%d%d",&a,&va);
            AddEdge(i,a,va);
        }
        int t=1;
        bfs(d1,t);
        bfs(d1,t);
        bfs(d2,t);
        for(int i=1;i<=n;++i)
        {printf("%d\n",max(d1[i],d2[i]));}
        for(int i=0;i<=n;++i)
        {G[i].clear();}
         edges.clear();
    }
    return 0;
}