Hướng giải của Đường đi lớn nhất trên cây
Nhớ rằng hướng dẫn giải này chỉ nên sử dụng khi bế tắc, và tuyệt đối không nên sao chép mã nguồn kèm theo. Hãy tôn trọng tác giả bài tập và người viết hướng dẫn giải.
Nộp mã nguồn lời giải chính thức trước khi giải bài tập đó có thể khiến bạn bị ban.
Nộp mã nguồn lời giải chính thức trước khi giải bài tập đó có thể khiến bạn bị ban.
Tác giả:
DP trên cây: DFS trả về đường đi dài nhất từ u xuống lá. Kết hợp hai nhánh con để cập nhật đáp án.
Solution Code
#include <bits/stdc++.h>
using namespace std;
const long long INF = -1e18;
long long ans = INF;
vector<vector<pair<int, long long>>> adj;
// DFS trả về đường đi dài nhất từ u xuống lá (không đi lên cha p)
long long dfs(int u, int p) {
long long best = 0; // Đường đi dài nhất từ u xuống 1 nhánh con
for (auto& [v, w] : adj[u]) if (v != p) {
long long child = dfs(v, u) + w; // Đường đi từ nhánh con v lên u
ans = max(ans, best + child); // Nối 2 nhánh con qua u
best = max(best, child);
}
ans = max(ans, best); // Trường hợp đường đi chỉ gồm 1 nhánh
return best;
}
int main() {
ios_base::sync_with_stdio(false); cin.tie(NULL);
int n; cin >> n; adj.resize(n + 1);
for (int i = 0; i < n - 1; i++) { int u, v; long long w; cin >> u >> v >> w; adj[u].push_back({v, w}); adj[v].push_back({u, w}); }
dfs(1, 0);
cout << ans << endl;
return 0;
}
Nhận xét