COJ 2003 选根 (树的重心)

我们可以用树形DP在线性复杂度内搞定重心。

 #include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#define PAU putchar(' ')
#define ENT putchar('\n')
using namespace std;
const int maxn=+;
struct tedge{int x,y,next;}adj[maxn<<];int ms=,fch[maxn];
void addedge(int u,int v){
adj[++ms]=(tedge){u,v,fch[u]};fch[u]=ms;
adj[++ms]=(tedge){v,u,fch[v]};fch[v]=ms;
return;
}
int s[maxn],f[maxn],cg,n;
void findcg(int u,int fa){
s[u]=;int mxs=;
for(int i=fch[u];i;i=adj[i].next){
int v=adj[i].y;if(v!=fa){
findcg(v,u);
s[u]+=s[v];
mxs=max(mxs,s[v]);
}
} f[u]=max(n-s[u],mxs);
if(f[u]<f[cg]) cg=u;
else if(f[u]==f[cg]) cg=min(u,cg);return;//编号小的
}
inline int read(){
int x=,sig=;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')sig=-;ch=getchar();}
while(isdigit(ch))x=*x+ch-'',ch=getchar();
return x*=sig;
}
inline void write(int x){
if(x==){putchar('');return;}if(x<)putchar('-'),x=-x;
int len=,buf[];while(x)buf[len++]=x%,x/=;
for(int i=len-;i>=;i--)putchar(buf[i]+'');return;
}
void init(){
n=read();f[cg=]=n;
for(int i=;i<n;i++) addedge(read(),read());
findcg(,);
write(cg);
return;
}
void work(){
return;
}
void print(){
return;
}
int main(){init();work();print();return ;}

搜索

复制

上一篇:python-廖雪峰,map/reduce学习笔记


下一篇:什么是ORM,以及在php上的使用?