Acwing 1072. 樹的最長路徑

題意：

每個邊有權值，求樹的直徑

題解：

兩遍dfs可以求，這里用樹形dp的方法，我們將1作為根節點來看這棵樹
我們可以將點看作是釘子，邊就是掛在釘子上的繩子，我們只需要求出第一長和次長的繩子，就可以得到最長的路徑
詳細過程不作表述

代碼：

#include<bits/stdc++.h> #define debug(a,b) printf("%s = %d\n",a,b); typedef long long ll; using namespace std;inline int read(){int s=0,w=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();//s=(s<<3)+(s<<1)+(ch^48);return s*w; } const int maxn=1e5+9; int ans; int n; vector<pair<int,int> >vec[maxn]; int dfs(int u,int fa){int dist=0;int d1=0,d2=0;for(int i=0;i<vec[u].size();i++){int v=vec[u][i].first;int w=vec[u][i].second;if(v==fa)continue;int d=dfs(v,u)+w;dist=max(dist,d);if(d>=d1)d2=d1,d1=d;else if(d>d2)d2=d;}ans=max(ans,d1+d2);return dist; } int main() {cin>>n;for(int i=1;i<n;i++){int u,v,w;cin>>u>>v>>w;vec[u].push_back(make_pair(v,w));vec[v].push_back(make_pair(u,w));}dfs(1,-1);cout<<ans<<endl;}

總結

以上是生活随笔為你收集整理的Acwing 1072. 树的最长路径的全部內容，希望文章能夠幫你解決所遇到的問題。

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