nowcoder 桃花 (两次DFS求直径)

桃花
题目描述

桃花一簇开无主，可爱深红映浅红。
——《题百叶桃花》
桃花长在桃树上，树的每个节点有一个桃花，调皮的HtBest想摘尽可能多的桃花。HtBest有一个魔法棒，摘到树上任意一条链上的所有桃花，由于HtBest法力有限，只能使用一次魔法棒，请求出Htbest最多可以摘到多少个桃花。

输入描述:

第一行有一个正整数n，表示桃树的节点个数。
接下来n-1行，第i行两个正整数ai,bi ，表示桃树上的节点ai,bi之间有一条边。

输出描述:

第一行一个整数，表示HtBest使用一次魔法棒最多可以摘到多少桃花。

示例1
输入

3
1 2
2 3

输出

3

示例2
输入

4
1 2
2 3
3 4

输出

4

备注

对于100%的测试数据：
1 ≤ n ≤ 1000000
数据量较大，注意使用更快的输入输出方式。

#include<iostream>
#include<cstring>
#include<stack>
#include<queue>
#include<cstdio>
#include<vector>
#include<string>
#include<map>
#include<unordered_map>
#include<algorithm>
#include<set>
using namespace std;


const int maxn = 1e6 + 50;
const int MOD = 1e9 + 7;
const double eps = 1e-9;


#define PI acos(-1)
#define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,x) for(int i=0;i<x;++i)
#define UP(i,x,y) for(int i=x;i<=y;i++) 
#define DOWN(i,x,y) for(int i=x;i>=y;i--)
#define DEBUG(x)  cout << "#: " << (x) << endl

using LL = long long;
using PII = pair<int,int>;
using VI = vector<int>;

VI G[maxn];
bool vis[maxn];
int n,ans,idx,cur;

void dfs(int u,int now){
    vis[u] = true;
    if(now > ans){
        ans = now;
        idx = u;
    }
    for(auto v : G[u]){
        if(!vis[v]){
            dfs(v,now + 1);
        }
    }
}   

int main(){
    IO;
    cin >> n;
    int u,v;
    for(int i = 0; i < n - 1; i++){
        cin >> u >> v;
        G[u].push_back(v); G[v].push_back(u);
    }
    ans = 0;
    dfs(1,0);
    ans = 0;
    mem(vis,false);
    dfs(idx,0);
    cout << ans + 1 << endl;
    return 0;
}