hihocoder #1224 : 赛车 dfs

#1224 : 赛车

Time Limit: 1 Sec

Memory Limit: 256 MB

题目连接

http://hihocoder.com/problemset/problem/1224

Description

幻想乡有一个赛车场。赛车场里有N个地点。同时地点之间还有单向的道路存在。

这些道路使得赛车场形成了一个外向树的结构。也就是说，道路将这N个地点连成了一个有根树。并且所有的边都是从父亲指向孩子的。

由于幽香喜欢刺激，每次她去赛车场都会从根节点出发，选择最长的一条路径来玩。

但是现在幽香感觉最长的路径还是太短了，她打算在赛车场里新建一条道路使得新的最长路径最长。

同时，如果道路形成了一个环，那么可能会出现交通事故，所以幽香新建的道路不能导致环的出现。

你能帮幽香算出新建一条道路后的最长路径吗？幽香知道根节点一定是1号点。

Input

一行一个数N，表示地点的数量。

接下来N-1行，每行两个数a和b，表示从点a到点b有一条单向路径。所有点从1到n标号。

数据范围：

n<=100000。

Output

一行表示新建一条边后的最长路径。

Sample Input

5
1 2
2 3
1 4
4 5

Sample Output

HINT

题意

题解：

http://media.hihocoder.com/contests/challenge14/hihoCoder-Challenge14-Solution.pdf

官方题解在上面

首先我们先dfs出最长链，然后我们正在dfs出每一个点能够最长延展多少

然后最后答案就是最长链的长度+最长延展多少就好了

代码：

//qscqesze

#include <cstdio>

#include <cmath>

#include <cstring>

#include <ctime>

#include <iostream>

#include <algorithm>

#include <set>

#include <bitset>

#include <vector>

#include <sstream>

#include <queue>

#include <typeinfo>

#include <fstream>

#include <map>

#include <stack>

typedef long long ll;

using namespace std;

//freopen("D.in","r",stdin);

//freopen("D.out","w",stdout);

#define sspeed ios_base::sync_with_stdio(0);cin.tie(0)

#define maxn 120051

#define mod 10007

#define eps 1e-9

int Num;

//const int inf=0x7fffffff;   //нчоч╢С

const int inf=~0u>>;

inline ll read()

{

    ll x=,f=;char ch=getchar();

    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}

    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}

    return x*f;

}

//**************************************************************************************

vector<int> E[maxn];

int fa[maxn];

int dep[maxn];

int key[maxn];

void dfs(int x,int deep)

{

    dep[x]=deep;

    for(int i=;i<E[x].size();i++)

    {

        int v=E[x][i];

        if(key[v])

            continue;

        fa[v]=x;

        dfs(v,deep+);

    }

}

int main()

{

    int n=read();

    for(int i=;i<n;i++)

    {

        int a=read(),b=read();

        E[a].push_back(b);

    }

    dfs(,);

    int mx=,bj;

    for(int i=;i<=n;i++)

    {

        if(dep[i]>mx)

        {

            mx=dep[i];

            bj=i;

        }

    }//找到最长链

    int ans=mx;

    int x = bj;

    while(x!=)

    {

        key[x]=;

        x=fa[x];

    }

    key[]=;

    for(int i=;i<=n;i++)

    {

        if(key[i])

        {

            dfs(i,);

        }

    }

    mx = ;

    for(int i=;i<=n;i++)

    {

        if(!key[i])

            mx = max(mx,dep[i]);

    }

    ans = max(,ans+mx-);

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

    return ;

}