1. 程式人生 > >大臣的旅費 兩次深搜求樹的直徑

大臣的旅費 兩次深搜求樹的直徑

很久以前,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;
}