树的直径（树形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;
}