树的直径（树形dp）

B4016 树的直径

题目描述

给定一棵 \(n\) 个结点的树，树没有边权。请求出树的直径是多少，即树上的最长路径长度是多少。

输入格式

第一行输入一个正整数 \(n\)，表示结点个数。

第二行开始，往下一共 \(n-1\) 行，每一行两个正整数 \((u,v)\)，表示一条边。

输出格式

输出一行，表示树的直径是多少。

输入输出样例 #1

输入 #1

5
1 2
2 4
4 5
2 3

输出 #1

3

说明/提示

数据保证，\(1 \leq n \leq 10^5\)。
更新d1 d2的顺序错了应该先更新d2再更新d1的

#include<iostream>
#include<vector>
using namespace std;
const int N=1e5+5;
vector<int>v[N];
int vis[N];
int ans=0;
int dfs(int x){
    vis[x]=1;
    int d1=0;
    int d2=0;
    for(int i=0;i<v[x].size();i++){
        if(!vis[v[x][i]]){
            int son=v[x][i];
            int s=dfs(son)+1;
            if(s>=d1)d2=d1,d1=s;
            else if(s>d2)d2=s;
        }
    }
    ans=max(ans,d1+d2);
    return d1;
}
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n-1;i++){
        int a,b;
        cin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    dfs(1);
    cout<<ans;
    return 0;
}