[POI2008]Sta
Description
给出一个N个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大
Input
给出一个数字N,代表有N个点.N<=1000000 下面N-1条边.
Output
输出你所找到的点,如果具有多个解,请输出编号最小的那个.
Sample Input
8
1 4
5 6
4 5
6 7
6 8
2 4
3 4
Sample Output
7
首先以1为根,然后记录size,sum,size记录有多少个儿子节点,sum记录答案,首先的更新就是sum[u]=sum[v]+size[v]
然后我们考虑旋根,当前根为u,要旋上来的节点为v,那么转移为
sum[x]=sum[x]-sum[son]-size[son];
size[x]=size[x]-size[son];
sum[son]=sum[son]+sum[x]+size[x];
size[son]=size[son]+size[x];
一定要按顺序,其实这个转移也不难想
/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline int read(){
int x=0,f=1;char ch=getchar();
for (;ch<'0'||ch>'9';ch=getchar()) if (ch=='-') f=-1;
for (;ch>='0'&&ch<='9';ch=getchar()) x=(x<<1)+(x<<3)+ch-'0';
return x*f;
}
inline void print(int x){
if (x>=10) print(x/10);
putchar(x%10+'0');
}
const int N=1e6;
int pre[(N<<1)+10],now[N+10],child[(N<<1)+10],size[N+10];
int tot,ID;
ll sum[N+10],Max;
void join(int x,int y){pre[++tot]=now[x],now[x]=tot,child[tot]=y;}
void build(int x,int fa){
size[x]=1;
for (int p=now[x],son=child[p];p;p=pre[p],son=child[p]){
if (son==fa) continue;
build(son,x);
size[x]+=size[son];
sum[x]+=sum[son]+size[son];
}
}
void work(int x,int fa){
if (Max<=sum[x]){
Max<sum[x]?ID=x:ID=min(ID,x);
Max=sum[x];
}
int size_x=size[x];
ll sum_x=sum[x];
for (int p=now[x],son=child[p];p;p=pre[p],son=child[p]){
if (son==fa) continue;
int size_son=size[son];
ll sum_son=sum[son];
sum[x]=sum[x]-sum[son]-size[son];
size[x]=size[x]-size[son];
sum[son]=sum[son]+sum[x]+size[x];
size[son]=size[son]+size[x];
work(son,x);
size[x]=size_x,sum[x]=sum_x;
size[son]=size_son,sum[son]=sum_son;
}
}
int main(){
int n=read();
for (int i=1;i<n;i++){
int x=read(),y=read();
join(x,y),join(y,x);
}
build(1,0);
work(1,0);
printf("%d\n",ID);
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步