洛谷 P3478 [POI2008]STA-Station

题目描述

The first stage of train system reform (that has been described in the problem Railways of the third stage of 14th Polish OI.

However, one needs not be familiar with that problem in order to solve this task.) has come to an end in Byteotia. The system consists of bidirectional segments of tracks that connect railway stations. No two stations are (directly) connected by more than one segment of tracks.

Furthermore, it is known that every railway station is reachable from every other station by a unique route. This route may consist of several segments of tracks, but it never leads through one station more than once.

The second stage of the reform aims at developing train connections.

Byteasar count on your aid in this task. To make things easier, Byteasar has decided that:

one of the stations is to became a giant hub and receive the glorious name of Bitwise, for every other station a connection to Bitwise and back is to be set up, each train will travel between Bitwise and its other destination back and forth along the only possible route, stopping at each intermediate station.

It remains yet to decide which station should become Bitwise. It has been decided that the average cost of travel between two different stations should be minimal.

In Byteotia there are only one-way-one-use tickets at the modest price of bythaler, authorising the owner to travel along exactly one segment of tracks, no matter how long it is.

Thus the cost of travel between any two stations is simply the minimum number of tracks segments one has to ride along to get from one stations to the other.

Task Write a programme that:

reads the description of the train system of Byteotia, determines the station that should become Bitwise, writes out the result to the standard output.

给出一个N个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大

输入输出格式

输入格式：

The first line of the standard input contains one integer () denoting the number of the railway stations. The stations are numbered from to . Stations are connected by segments of tracks. These are described in the following lines, one per line. Each of these lines contains two positive integers and (), separated by a single space and denoting the numbers of stations connected by this exact segment of tracks.

输出格式：

In the first and only line of the standard output your programme should print out one integer - the optimum location of the Bitwise hub.

If more than one optimum location exists, it may pick one of them arbitrarily.

输入样例：

输出样例：

 //本题相当于洛谷P2986 [USACO10MAR]伟大的奶牛聚集   而且是简化版的题目。。。

 #include<bits/stdc++.h>

 using namespace std;

 #define ll long long

 #define man 1000010

 ll son[man],dis[man],n,f[man];

 struct edge

 {    ll next,to;}e[man<<];

 ll head[man<<],num=;

 inline void add(ll from,ll to)

 {    e[++num].next=head[from];e[num].to=to;head[from]=num;}

 ll precalc(int u,int fa)

 {    int tot=;

     for(ll i=head[u];i;i=e[i].next)

     {    ll to=e[i].to;

         if(to==fa) continue;

         ll child=precalc(to,u);

         dis[u]+=dis[to]+child;

         tot+=child;

         }

     return son[u]=tot+;

     }

 void dfs(int u,int fa)

 {    for(int i=head[u];i;i=e[i].next)

     {    ll to=e[i].to;

         if(to==fa)continue;

         f[to]=f[u]-son[to]*+(n-son[to])*;

         dfs(to,u);

         }

     }

 void ad(int elem)

 {    elem+=dis[];}

 int main()

 {    scanf("%lld",&n);

     for(int i=;i<n;i++)

     {    ll u,v;

         scanf("%lld%lld",&u,&v);

         add(u,v);add(v,u);

         }

     memset(f,,sizeof(f));

     precalc(,-);//预处理每个节点的子节点的个数和每个子树的距离

     dfs(,-);

     for_each(f+,f+n+,ad);//相当于f[i]+=dis[1];

     cout<<(max_element(f+,f+n+)-f)<<endl;

     return ;

     }