hihoCoder #1050 : 树中的最长路

 

题意：

　　求出树上最长路径的长度，并返回。

 

思路：

　　刚看到数据<=10^5，假如是单分支的树，那么有5万层，就不能递归，那就用桟实现， 那就要将长度信息保存在另开的数组中，很麻烦！！这题专门给递归实现的。DFS到叶子开始返回长度，经过每个节点就计算以该结点为转折点的最长路径，直到根为止，就找到最大路径了。

 

 

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
vector< vector<int> > gra;    //树以临接表记录（有向图）
int maxsum;
int DFS(int num)
{
    if(gra[num].empty()==true)    //出口
        return 1;
    int tmp, big1=0, big2=0;    //多叉树，big1和2分别是两条到本结点的最长路径。
    for(int i=0; i<gra[num].size(); i++)
    {
        tmp=DFS( gra[num][i]);            
        if(tmp>big1)
        {
            big2=big1;
            big1=tmp;
        }
        else if(tmp>big2)
            big2=tmp;
    }
    maxsum=max(maxsum, big1+big2);    //计算以本节点为转折点的最长路径。
    return big1+1;
}

int main()
{
    //freopen("input.txt", "r", stdin);
    int t, i, tmp1, tmp2;
    cin>>t;
    gra.resize(t);
    for(i=1; i<t; i++)
    {
        scanf("%d%d", &tmp1, &tmp2);//用c读入会省一半时间
        gra[--tmp1].push_back(--tmp2);
    }
    DFS(0);
    printf("%d\n", maxsum);
    return 0;
}