hiho#1050 : 树中的最长路

1050 : 树中的最长路

经典算法。

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;


int n;
const int N=100007;
vector<int>ve[N];
int dis[N];
int vis[N];

void dfs(int now,int len)
{
    if(vis[now])
        return ;

    dis[now]=len;
    vis[now]=1;
    for(int i=0;i<ve[now].size();i++)
    {
        int to=ve[now][i];
        dfs(to,len+1);
    }

}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        memset(vis,0,sizeof vis);
        memset(dis,0,sizeof dis);
        for(int i=0;i<=n;i++)
            ve[i].clear();

        for(int i=0;i<n-1;i++)
        {

            int a,b;
            scanf("%d%d",&a,&b);
            ve[a].push_back(b);
            ve[b].push_back(a);

        }
        dfs(1,0);
        int ma=1;
        for(int i=1;i<=n;i++)
        if(dis[ma]<dis[i])
            ma=i;
        memset(vis,0,sizeof vis);
        dfs(ma,0);
        ma=0;

        for(int i=1;i<=n;i++)
            ma=max(ma,dis[i]);

        printf("%d\n",ma);


    }



    return 0;
}