acwing 1072. 树的最长路径（树形dp）

传送门

描述

给定一棵树，树中包含 n 个结点（编号1~n）和 n−1 条无向边，每条边都有一个权值。

现在请你找到树中的一条最长路径。

换句话说，要找到一条路径，使得使得路径两端的点的距离最远。

注意：路径中可以只包含一个点。

输入格式

第一行包含整数 n。

接下来 n−1 行，每行包含三个整数 ai,bi,ci，表示点 ai 和 bi 之间存在一条权值为 ci 的边。

输出格式

输出一个整数，表示树的最长路径的长度。

数据范围

1≤n≤10000,
1≤ai,bi≤n,
−105≤ci≤105

输入样例：

6
5 1 6
1 4 5
6 3 9
2 6 8
6 1 7

输出样例：

22

这道题我们首先考虑建边的时候建无向边，然后我们根节点任取一个都行，然后对于每一个节点找到他的子节点分别的最大长度，然后取两个最

大的，对于每一个节点我们的答案都要进行更新，我们设m1是当前节点子节点的最大值，m2是当前节点的次大值，然后就有

ans=max（ans，m1+m2），这样就能保证我们找到的答案是最大值，然后返回给上一层的最大值就是m1

注意，m1是返回给上层的最大值，而ans是我们答案的最大值，因为不一定点多距离就大

#####AC代码如下：

#include<bits/stdc++.h>
using namespace std;
const int maxn=200010;
int h[maxn],e[maxn],ne[maxn],w[maxn];
int n,m,cnt,ans=0;
bool vis[maxn];
void add(int x,int y,int p) {
	w[cnt]=p,e[cnt]=x,ne[cnt]=h[y],h[y]=cnt++;
}
int dfs(int x) {
	int res=0,m1=0,m2=0;
	vis[x]=true;
	for(int i=h[x]; i!=-1; i=ne[i]) {
		int y=e[i];
		if(!vis[y]) {
			int s=dfs(y);
			s+=w[i];
			if(s>m1)swap(m1,s);
			if(s>m2)swap(m2,s);
		}
	}
	ans=max(m1+m2,ans);
	return m1;
}
int main() {
	cin>>n;
	for(int i=0; i<maxn; i++)h[i]=-1;
	for(int i=1; i<n; i++) {
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
		add(b,a,c);
	}
	dfs(1);
	cout<<ans<<endl;
	return 0;
}