树的最长路径（树形dp）

题意描述：
给定一棵树，树中包含 n 个结点（编号1~n）和 n−1 条无向边，每条边都有一个权值。
现在请你找到树中的一条最长路径。
换句话说，要找到一条路径，使得使得路径两端的点的距离最远。
注意：路径中可以只包含一个点。
输入格式
第一行包含整数 n。
接下来 n−1 行，每行包含三个整数 ai,bi,ci，表示点 ai 和 bi 之间存在一条权值为 ci 的边。
输出格式
输出一个整数，表示树的最长路径的长度。
数据范围
1≤n≤10000,
1≤ai,bi≤n,
−1e5≤ci≤1e5
输入样例：

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

输出样例：
 

22

输入样例：

6 
3 1 1000
1 4 10
4 2 100
4 5 50
4 6 100

输出样例：
 

1110

对于树上的每一个节点都求出它所有子树的最大深度和第二大深度，两者相加便是以该节点为根节点的子树的最大链长

vector 邻接表

#include<bits/stdc++.h>
using namespace std; using ll = long long;
int dir[4][2] = { 1, 0, -1, 0, 0, 1, 0, -1 };
//using lll = __int128; template <class T> istream& read(T& x, istream& cin = std::cin) { T num = 0; bool f = 0; char ch = 0; while (!isdigit(ch)) { f |= ch == '-'; if (!cin.get(ch)) return cin; }while (isdigit(ch)) { num = (num << 3) + (num << 1) + (ch ^ 48); if (!cin.get(ch)) break; }x = f ? -num : num; return cin; }template <class T> ostream& write(T x, ostream& cout = std::cout) { if (x < 0) cout.put('-'), x = -x; if (x > 9) write(x / 10); cout.put(x % 10 + '0'); return cout; }ostream& operator<<(ostream& cout, lll x) { write(x); return cout; }istream& operator>>(istream& cin, lll& x) { return read(x); }bool check(int i, int j);
bool check(int i, int j);
using pii = pair<int, int>;

const int N = 1e5 + 10, mod = 1e9 + 7, INF = 0x3f3f3f3f;

int n, m, ans;

vector<pii> g[N];

void init() {
	cin >> n;
	for (int i = 0; i < n - 1; i++) {
		int x, y, z;
		cin >> x >> y >> z;
		g[x].emplace_back(y, z);
		g[y].emplace_back(x, z);
	}
	return;
}

int dfs(int u, int pre) {
	int d1 = 0, d2 = 0;  //u的子树的最大长度，次大长度
	for (auto& [nx, w] : g[u]) {
		if (nx == pre) continue;
		int d = dfs(nx, u) + w;
		if (d >= d1) {
			d2 = d1;
			d1 = d;
		}
		else if (d > d2) {
			d2 = d;
		}
	}
	ans = max(d1 + d2, ans);
	return d1;
}

void solve() {
	dfs(1, -1);
	cout << ans;
	return;
}

int main(void) {
	ios::sync_with_stdio(0); cin.tie(0); cout << setprecision(6) << fixed;
	int TT = 1;
	//cin >> TT;
	for (int ii = 1; ii <= TT; init(), solve(), ii++, cout << "\n") {}
	return 0;
}

数组模拟邻接表

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
int dir[4][2] = { 1, 0, -1, 0, 0, 1, 0, -1 };
//using lll = __int128;template <class T> istream& read(T& x, istream& cin = std::cin){T num = 0;bool f = 0;char ch = 0;while (!isdigit(ch)) {f |= ch == '-';if (!cin.get(ch)) return cin;}while (isdigit(ch)){num = (num << 3) + (num << 1) + (ch ^ 48);if (!cin.get(ch)) break;}x = f ? -num : num;return cin;}template <class T> ostream& write(T x, ostream& cout = std::cout) {if (x < 0) cout.put('-'), x = -x;if (x > 9) write(x / 10);cout.put(x % 10 + '0');return cout;}ostream& operator<<(ostream& cout, lll x) {write(x);return cout;}istream& operator>>(istream& cin, lll &x) {return read(x);}bool check(int i, int j);
bool check(int i, int j);

const int N = 1e5 + 10, mod = 1e9 + 7, INF = 0x3f3f3f3f;

int n, m, h[N], ne[N], e[N], idx, w[N], ans;

void add(int x, int y, int z) {
	e[idx] = y, ne[idx] = h[x], w[idx] = z, h[x] = idx ++;
}

void init() {
	memset(h, -1, sizeof h);
	cin >> n;
	for (int i = 0; i < n - 1; i ++) {
		int x, y, z;
		cin >> x >> y >> z;
		add(x, y, z);
		add(y, x, z);
	}
	//	cout << "初始化完成\n";
	return;
}

int dfs(int u, int pre) {
	int d1 = 0, d2 = 0;
	for (int i = h[u]; i != -1; i = ne[i]) {
		int j = e[i];
		if (j == pre) continue;
		int d = dfs(j, u) + w[i];
		if (d >= d1) {
			d2 = d1;
			d1 = d;
		} else if (d > d2) {
			d2 = d;
		}
	}
	ans = max(ans, d1 + d2);
	return d1;
}

void solve() {
	dfs(1, -1);
	cout << ans;
	return;
}

int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout << setprecision(6) << fixed;//保留6位小数从小数点后计数
	int TT = 1;
	//cin >> TT;
	for (int ii = 0; ii < TT; init(), solve(), ii++, cout << "\n") {}
	return 0;
}