bzoj3391[Usaco2004 Dec]Tree Cutting网络破坏*

bzoj3391[Usaco2004 Dec]Tree Cutting网络破坏

题意：

给一棵树，问去掉哪个点后可以使剩下的每个子树大小都小于等于节点总数的一半。n≤10000。

题解：

dfs的时候求一下子树大小，以当前节点的父节点为根节点的子树大小为n-以当前节点为根节点的子树大小。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#define inc(i,j,k) for(int i=j;i<=k;i++)
#define maxn 10010
using namespace std;

inline int read(){
    char ch=getchar(); int f=1,x=0;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return f*x;
}
struct e{int t,n;}es[maxn*2]; int ess,g[maxn];
void pe(int f,int t){
    es[++ess]=(e){t,g[f]}; g[f]=ess; es[++ess]=(e){f,g[t]}; g[t]=ess;
}
int sz[maxn],n,ans[maxn],tot;
void dfs(int x,int fa){
    int mx=0; sz[x]=1; for(int i=g[x];i;i=es[i].n)if(es[i].t!=fa){
        dfs(es[i].t,x); sz[x]+=sz[es[i].t]; mx=max(mx,sz[es[i].t]);
    }
    mx=max(mx,n-sz[x]); if(mx<=n/2)ans[++tot]=x;
}
int main(){
    n=read(); inc(i,1,n-1){int x=read(),y=read(); pe(x,y);} dfs(1,0);
    if(!tot){printf("NONE"); return 0;}
    sort(ans+1,ans+1+tot); inc(i,1,tot)printf("%d\n",ans[i]); return 0;
}

 

20161114