【poj1655】Balancing Act

来源:互联网 发布:电火花电脉冲编程 编辑:程序博客网 时间:2024/06/07 06:20

http://poj.org/problem?id=1655 (题目链接)

题意:给出一棵树,求树上一点使得使得删除一点后该树的最大子树最小。

solution
  树的重心裸题。
  随意取一点作为根节点,dfs维护当前节点的最大子树大小以及它的父亲子树大小,更新答案。

代码:

// poj1655#include<algorithm>#include<iostream>#include<cstdlib>#include<cstring>#include<cstdio>#include<cmath>#define LL long long#define inf 2147483640#define Pi acos(-1.0)#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);using namespace std;const int maxn=20010;struct edge {int to,next;}e[maxn<<2];int head[maxn],vis[maxn],son[maxn],cnt,size,ans,n;void insert(int u,int v) {    e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;    e[++cnt].to=u;e[cnt].next=head[v];head[v]=cnt;}void dfs(int x) {    vis[x]=1,son[x]=0;    int tmp=0;    for (int i=head[x];i;i=e[i].next) if (!vis[e[i].to]) {            dfs(e[i].to);            son[x]+=son[e[i].to]+1;            tmp=max(tmp,son[e[i].to]+1);        }    tmp=max(tmp,n-son[x]-1);    if (tmp<size || (tmp==size && x<ans)) size=tmp,ans=x;}int main() {    int T;scanf("%d",&T);    while (T--) {        scanf("%d",&n);        memset(vis,0,sizeof(vis));memset(head,0,sizeof(head));        cnt=ans=0;size=inf;        for (int i=1;i<n;i++) {            int u,v;            scanf("%d%d",&u,&v);            insert(u,v);        }        dfs(1);        printf("%d %d\n",ans,size);    }    return 0;}
0 0
原创粉丝点击