http://www.lydsy.com/JudgeOnline/problem.php?id=1131
dp[i]=dp[fa[i]]-son[i]+n-son[i]
#include<cstdio> #include<iostream> using namespace std; #define N 1000001 int front[N],to[N<<1],nxt[N<<1],tot; int fa[N]; bool vis[N]; int dep[N],son[N]; long long dp[N]; int head=1,tail=1; int q[N]; void read(int &x) { x=0; char c=getchar(); while(!isdigit(c)) c=getchar(); while(isdigit(c)) { x=x*10+c-'0'; c=getchar(); } } void add(int u,int v) { to[++tot]=v; nxt[tot]=front[u]; front[u]=tot; to[++tot]=u; nxt[tot]=front[v]; front[v]=tot; } void bfs() { q[1]=1; vis[1]=true; int now,t; while(head<=tail) { now=q[head++]; for(int i=front[now];i;i=nxt[i]) { t=to[i]; if(!vis[t]) { vis[t]=true; fa[t]=now; q[++tail]=t; } } } } void dfs(int x,int d) { dep[x]=d; son[x]=1; for(int i=front[x];i;i=nxt[i]) if(!dep[to[i]]) { dfs(to[i],d+1); son[x]+=son[to[i]]; } } int main() { int n; read(n); int u,v; for(int i=1;i<n;++i) { read(u); read(v); add(u,v); } bfs(); dfs(1,1); for(int i=1;i<=n;++i) dp[1]+=dep[i]; int ans=1; long long mx=dp[1]; for(int i=2;i<=n;++i) { u=q[i]; dp[u]=dp[fa[u]]-son[u]+n-son[u]; if(dp[u]>mx) { mx=dp[u]; ans=u; } else if(dp[u]==mx && u<ans) ans=u; } cout<<ans; }
题目描述
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.