#include <iostream> #include <cstring> using namespace std;//maxv：源點能到的最遠點，maxdis:最遠點對應的距離, const int maxn = 1e4 + 5; struct Edge { int to, next, w; }edges[2 * maxn]; int head[maxn], maxdis,maxv, tot; void add(int u, int v, int w) {edges[tot] = { v, head[u], w };head[u] =tot++; }void dfs(int u, int f, int Val) {if (maxdis < Val){maxdis = Val;maxv = u;}for (int e = head[u]; e != -1; e = edges[e].next) {int v = edges[e].to, w = edges[e].w;if (v == f) continue; //父節點已經訪問過，防止重復遍歷，相反孩子不會重復遍歷。dfs(v, u, Val + w);} }int main() {int e, u, v, w, s;cin >> e;memset(head, -1, sizeof(head));for (int i = 1; i <= e; i++) {cin >> u >> v >> w;add(u, v, w), add(v, u, w);}dfs(1, -1, 0); //從結點1開始遍歷，找到最遠點maxv及對應的最遠距離maxdismaxdis = 0;cout <<maxv<<endl;//輸出直徑的第一個端點dfs(maxv, -1, 0);//從結點maxv開始遍歷，找到最遠點對應的距離maxdiscout << maxdis << endl; //輸出樹的直徑cout <<maxv<<endl;//輸出樹的直徑的第二個端點return 0; } #include <cstdio> #include <cstring> #include <algorithm> using namespace std; int const MAX = 100005; int head[MAX], dp[MAX][2]; int n, s, cnt, ans; struct EDGE {int v, w, next; } e[MAX]; void Add(int u, int v, int w) {e[cnt].v = v;e[cnt].w = w;e[cnt].next = head[u];head[u] = cnt++; } void DFS(int u, int fa) {dp[u][0] = dp[u][1] = 0;for (int i = head[u]; i != -1; i = e[i].next){int v = e[i].v;int w = e[i].w;if (v != fa){DFS(v, u);if (dp[u][0] < dp[v][0] + w){int tmp = dp[u][0];dp[u][0] = dp[v][0] + w;dp[u][1] = tmp;}else if (dp[u][1] < dp[v][0] + w)dp[u][1] = dp[v][0] + w;}}ans = max(ans, dp[u][1] + dp[u][0]);return; } int main() {cnt = 0;ans = 0;memset(head, -1, sizeof(head));scanf("%d %d", &n, &s);int sum = 0;for (int i = 0; i < n - 1; i++){int u, v, w;scanf("%d %d %d", &u, &v, &w);Add(u, v, w);Add(v, u, w);sum += 2 * w;}DFS(s, -1);printf("%d\n", ans); }

?

總結

以上是生活随笔為你收集整理的图论--树的直径--DFS+树形DP模板的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。