[POJ1655]Balancing Act（树形dp）

题目描述

传送门
题意：一个点的balance值定义为删除这个点之后形成的子树节点个数的最大值。求一颗树种点的balance值的最小值。

题解

dfs求子树的大小即可。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int max_n=2e4+5;
const int max_e=max_n*2;
const int INF=2e9;
int T,n,ans=INF,ansp,x,y;
int tot,point[max_n],next[max_e],v[max_e],size[max_n];

inline void clear(){
    n=ansp=tot=0; ans=INF;
    memset(size,0,sizeof(size));memset(point,0,sizeof(point));memset(next,0,sizeof(next));memset(v,0,sizeof(v));
}
inline void add(int x,int y){++tot;next[tot]=point[x];point[x]=tot;v[tot]=y;}
inline void dfs(int x,int fa){
    size[x]=1; int ansnow=0,ansnowp=0;
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        dfs(v[i],x);
        size[x]+=size[v[i]];
        if (size[v[i]]>ansnow){
            ansnow=size[v[i]];
            ansnowp=x;
        }
      }
    if (n-size[x]>ansnow) ansnow=n-size[x],ansnowp=x;
    if (ansnow<ans) ans=ansnow,ansp=ansnowp;
}
int main(){
    scanf("%d",&T);
    while (T--){
        clear();
        scanf("%d",&n);
        for (int i=1;i<n;++i) scanf("%d%d",&x,&y),add(x,y),add(y,x);
        dfs(1,0);
        printf("%d %d\n",ansp,ans);
    }
}