历届试题 大臣的旅费 树形DP
生活随笔
收集整理的這篇文章主要介紹了
历届试题 大臣的旅费 树形DP
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
????? 題目鏈接:大臣的旅費
??? 思路:錦囊說用廣搜,可惜這題沒說數據范圍,擔心復雜度太高,我就直接用的樹形DP--求樹的最遠路徑。
??? 以城市1為整棵樹的根結點,d(i)表示以i為根結點的子樹的最遠路徑,還有一個f(i)表示以i為根結點,從根節點i出發的最遠距離,狀態轉移方程d(i) = max1 + max2,或則d(i) = max,即選取最長的兩條子樹的最遠路徑相加,或只有一個子節點就是加上最大值。f(i) = max(len(i, j) + f(j));
AC代碼:
#include <cstdio> #include <cmath> #include <algorithm> #include <cstring> #include <utility> #include <string> #include <iostream> #include <map> #include <set> #include <vector> #include <queue> #include <stack> using namespace std; #define eps 1e-10 #define inf 0x3f3f3f3f #define PI pair<int, int> const int maxn = 1e4 + 5;vector<PI>son[maxn]; int d[maxn]; bool cmp(int a, int b) {return a > b; }int dfs(int u, int pre) {int n = son[u].size();d[u] = 0;int ans = 0;vector<int>road;if(!n) return 0;for(int i = 0; i < n; ++i) {int v = son[u][i].first, cost = son[u][i].second;if(v == pre) continue;int h = dfs(v, u);ans =max(ans, h + cost);road.push_back(h + cost);}sort(road.begin(), road.end(), cmp);if(road.size() == 1) d[u] = road[0];else if(road.size() > 1)d[u] = road[0] + road[1];return ans; }int main() {int n;while(scanf("%d", &n) == 1) {memset(d, 0, sizeof(d));for(int i = 0; i < n; ++i) son[i].clear();int x, y, cost;for(int i = 1; i < n; ++i) {scanf("%d%d%d", &x, &y, &cost);son[x-1].push_back(make_pair(y-1, cost));son[y-1].push_back(make_pair(x-1, cost));}dfs(0, -1);int ans = 0;for(int i = 0; i < n; ++i) {ans = max(ans, d[i]);}printf("%d\n", (ans + 1)*ans/2 + ans * 10);}return 0; }
如有不當之處歡迎指出!
轉載于:https://www.cnblogs.com/flyawayl/p/8305410.html
總結
以上是生活随笔為你收集整理的历届试题 大臣的旅费 树形DP的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: CloudStack学习-2
- 下一篇: 8.Layout布局应用