hihocoder-第十一周 树中的最长路

题意：找出二叉树中相距最远的两个节点的距离。

输入

每个测试点（输入文件）有且仅有一组测试数据。

每组测试数据的第一行为一个整数N，意义如前文所述。

每组测试数据的第2~N行，每行分别描述一根木棍，其中第i+1行为两个整数Ai，Bi，表示第i根木棍连接的两个小球的编号。

对于20%的数据，满足N<=10。

对于50%的数据，满足N<=10^3。

对于100%的数据，满足N<=10^5，1<=Ai<=N, 1<=Bi<=N

小Hi的Tip：那些用数组存储树边的记得要开两倍大小哦！

输出

对于每组测试数据，输出一个整数Ans，表示给出的这棵树中距离最远的两个结点之间相隔的距离。

样例输入

8
1 2
1 3
1 4
4 5
3 6
6 7
7 8

样例输出

6

递归求解，对于每一个树或者子树来说，最长的距离是一棵子树中最长的距离l1(+1)加上另一棵子树最长的距离l2(+1)（ maxLen = l1 + l2 + 2)

#include <iostream>
#include <vector>
using namespace std;

vector< vector<int> > g(100005);
int ans = 0;

int dfs(int pre, int cur){
    int l1 = -1, l2 = -1;                   //l1为子树中最大长度，l2为子树中第二大长度
    for(int i = 0; i < g[cur].size(); i++){  //枚举每一个子树（子路径）
        if(g[cur][i] == pre) continue;       //每一个子树（子路径）都只能算一遍，保证不重复走
        int l = dfs(cur, g[cur][i]);
        if(l > l1) {l2 = l1; l1 = l;}       //更新长度
        else l2 = max(l, l2);
    }
    ans = max(ans, l1+l2+2);              //对于每一棵子树的根节点都更新一次最终的最大长度
    return l1+1;
}

int main()
{
    int N;
    cin >> N;
    int a, b;
    for(int i = 1; i < N; i++){
        cin >> a >> b;
        g[a-1].push_back(b-1);
        g[b-1].push_back(a-1);
    }
    dfs(-1, 0);
    cout << ans << endl;
    return 0;
}