藍橋杯 PREV-9 大臣的旅費 (兩次DFS)
阿新 • • 發佈:2019-02-03
題意:求樹中最遠的兩個點的距離。
解題方案:第一遍dfs,任取一個點,找到離它最遠的點a,第二遍dfs,以a為起點,找到離a最遠的點b,ab即為樹中最遠的兩個點。證明:樹的直徑(最長路)的詳細證明。要注意到一點:樹中的任意兩個點一定連通,且路徑唯一。
#include <iostream> #include <cstdio> #include <algorithm> #include <cstring> #include <string> #include <cmath> #include <vector> #include <queue> #include <stack> #include <set> #include <map> using namespace std; #define FOR(i,k,n) for(int i=k;i<n;i++) #define FORR(i,k,n) for(int i=k;i<=n;i++) #define scan(a) scanf("%d",&a) #define scann(a,b) scanf("%d%d",&a,&b) #define scannn(a,b,c) scanf("%d%d%d",&a,&b,&c) #define mst(a,n) memset(a,n,sizeof(a)) #define ll long long #define N 10005 #define mod 1000000007 #define INF 0x3f3f3f3f const double eps=1e-8; const double pi=acos(-1.0); int n,ans,start; int vis[N]; vector<pair<int,int> > E[N]; void Dfs(int u,int len) { int flag=0; for(int i=0;i<E[u].size();i++) { int v=E[u][i].first; int d=E[u][i].second; if(!vis[v]) { flag=1; vis[v]=1; Dfs(v,len+d); vis[v]=0; } } if(!flag) { if(ans<len) ans=len,start=u; } } int main() { //freopen("in.txt","r",stdin); //freopen("out.txt","w",stdout); scan(n); FOR(i,0,n-1) { int a,b,c; scannn(a,b,c); E[a].push_back(make_pair(b,c)); E[b].push_back(make_pair(a,c)); } ans=0; mst(vis,0); vis[1]=1; Dfs(1,0); ans=0; mst(vis,0); vis[start]=1; Dfs(start,0); //printf("%d\n",ans); printf("%d\n",ans*10+ans*(ans+1)/2); return 0; }