hihocoder#1050 : 树中的最长路（树中最长路算法 两次BFS找根节点求最长+BFS标记路径长度+bfs不容易超时，用dfs做TLE了）

#include <iostream>
#include <string>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include <algorithm>
#include <vector>
#include <queue>
#define N 100000+5

using namespace std;
int n, tail; //用tail来标记第一次bfs找到的叶子节点
bool vis[N];
int fa[N]; //用来记录路径长度的数组
vector<int>q[N];

void BFS_root(int s)
{
    int i, j;
    int len;
    queue<int>p;
    while(!p.empty())
        p.pop();
    p.push(s);
    vis[s]=true;
    while(!p.empty())
    {
        int dd=p.front();
        p.pop();
        len=q[dd].size();
        for(i=0; i<len; i++)
        {
            if(vis[q[dd][i]]==false )
            {
                fa[q[dd][i]]=fa[dd]+1;
                vis[q[dd][i]]=true;
                p.push(q[dd][i]);
            }
        }
    }
    //
    tail=1;
    int ff=fa[1];
    for(j=2; j<=n; j++)
    {
        if(fa[j]>ff)
        {
            ff=fa[j]; tail=j;
        }
    }
}
void BFS_len_tree(int s)
{
    int i, j;
    int len;
    queue<int>p;
    while(!p.empty())
        p.pop();
    vis[s]=true;
    p.push(s);
    while(!p.empty())
    {
        int dd=p.front();
        p.pop();
        len=q[dd].size();
        for(i=0; i<len; i++)
        {
            if(vis[q[dd][i]]==false )
            {
                fa[q[dd][i]]=fa[dd]+1;
                vis[q[dd][i]]=true;
                p.push(q[dd][i]);
            }
        }
    }
    int ans=fa[s];
    for(j=1; j<=n; j++)
    {
        if(fa[j]>ans)
            ans=fa[j];
    }
    printf("%d\n", ans);
}

int main()
{
    scanf("%d", &n);
    int i, j;
    int u, v;
    for(i=0; i<=n; i++)
        q[i].clear();
    for(i=0; i<n-1; i++)
    {
        scanf("%d %d", &u, &v);
        q[u].push_back(v);
        q[v].push_back(u);
    }
    memset(vis, false, sizeof(vis));
    memset(fa, 0, sizeof(fa));
    BFS_root(1);

    memset(vis, false, sizeof(vis));
    memset(fa, 0, sizeof(fa));
    BFS_len_tree(tail);

    return 0;
}