poj 1655 Balancing Act (树形dp)

来源：互联网发布：软件远程升级方案编辑：程序博客网时间：2024/05/21 14:53

对于每个节点，它的balance要么来自于父亲节点，要么来自于儿子节点。
定义dp[u]为节点u的balance，sum[u]为以u为根的子树的节点个数，n为总共的节点个数。
如果来自于父亲节点，则dp[u]=n-sum[u];
如果来自于儿子节点，则dp[u]=sum[v];

#include<cstdio>#include<cstring>#include<algorithm>#include<cstdlib>#include<vector>#define N 20005using namespace std;vector<int>edge[N];int sum[N],dp[N],n;void dfs(int u,int fa){    int i,len=edge[u].size(),v;    sum[u]=1;    dp[u]=0;    for(i=0;i<len;i++){        v=edge[u][i];        if(v==fa) continue;        dfs(v,u);        dp[u]=max(dp[u],sum[v]);        sum[u]+=sum[v];    }    dp[u]=max(dp[u],n-sum[u]);}int main(){    int t;    scanf("%d",&t);    while(t--){        int i;        scanf("%d",&n);        for(i=0;i<=n;i++) edge[i].clear();        for(i=1;i<n;i++){            int a,b;            scanf("%d%d",&a,&b);            edge[a].push_back(b);            edge[b].push_back(a);        }        dfs(1,0);        int ans=99999999,si;        for(i=1;i<=n;i++){            if(dp[i]<ans){                ans=dp[i];                si=i;            }        }        printf("%d %d\n",si,ans);    }    return 0;}

0 0