大臣的旅費 兩次深搜求樹的直徑
阿新 • • 發佈:2019-01-14
很久以前,T王國空前繁榮。為了更好地管理國家,王國修建了大量的快速路,用於連線首都和王國內的各大城市。
為節省經費,T國的大臣們經過思考,制定了一套優秀的修建方案,使得任何一個大城市都能從首都直接或者通過其他大城市間接到達。同時,如果不重複經過大城市,從首都到達每個大城市的方案都是唯一的。
J是T國重要大臣,他巡查於各大城市之間,體察民情。所以,從一個城市馬不停蹄地到另一個城市成了J最常做的事情。他有一個錢袋,用於存放往來城市間的路費。
聰明的J發現,如果不在某個城市停下來修整,在連續行進過程中,他所花的路費與他已走過的距離有關,在走第x千米到第x+1千米這一千米中(x是整數),他花費的路費是x+10這麼多。也就是說走1千米花費11,走2千米要花費23。
#include<bits/stdc++.h> using namespace std; #define MAXN 10100 struct Edge { int v,w; Edge(int vv,int ww):v(vv),w(ww) {} }; int n; int dist[MAXN],max_len,End; vector<vector<Edge> >G; void dfs(int u,int father,int len) {//起點,父親節點,兩者之間長度 if(len>max_len)max_len=len,End=u;//更新終點; for(int i=0; i<G[u].size(); i++) {//對於每一個節點,進行深搜; int v=G[u][i].v,w=G[u][i].w;//與之相連的點; if(v==father)continue; dist[v]=max(dist[v],len+w);//更新到子節點之間的最大距離; dfs(v,u,len+w);//繼續深搜; } } int main() { int u,v,w; while(~scanf("%d",&n)) { G.clear();//清空 G.resize(n+2);//申請空間 for(int i=1; i<n; i++) { scanf("%d%d%d",&u,&v,&w); G[u].push_back(Edge(v,w)); G[v].push_back(Edge(u,w)); //無向圖; } memset(dist,0,sizeof(dist)); max_len=0; dfs(1,-1,0);//第一次深搜找到距離某點最遠的點op dfs(End,-1,0);//求距離op最遠的點,兩次深搜求出直徑; int ans=0; for(int i=1; i<=n; i++) { ans=max(ans,dist[i]); //其實可以直接是dist[End] } //cout<<"ans"<<ans<<" "<<dist[End]<<endl; int tt=0; for(int i=1; i<=ans; i++) tt+=(10+i);//求出花費; printf("%d\n",tt); } return 0; }