树形dp求树的重心

Balancing Act http://poj.org/problem?id=1655

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 #define mt(a,b) memset(a,b,sizeof(a))
 using namespace std;
 const int M=;
 vector<int> g[M];
 int dp[M],num[M],n;
 void dfs(int u,int fa){
     dp[u]=;
     num[u]=;
     int len=g[u].size();
     for(int i=;i<len;i++){
         int v=g[u][i];
         if(v!=fa){
             dfs(v,u);
             num[u]+=num[v];
             dp[u]=max(dp[u],num[v]);
         }
     }
     dp[u]=max(dp[u],n-num[u]);
 }
 int main(){
     int t;
     while(~scanf("%d",&t)){
         while(t--){
             scanf("%d",&n);
             for(int i=;i<=n;i++) g[i].clear();
             for(int i=,u,v;i<n-;i++){
                 scanf("%d%d",&u,&v);
                 g[u].push_back(v);
                 g[v].push_back(u);
             }
             dfs(,-);
             int sma=n;
             for(int i=;i<=n;i++){
                 sma=min(sma,dp[i]);
             }
             for(int i=;i<=n;i++){
                 if(dp[i]==sma){
                     printf("%d %d\n",i,dp[i]);
                     break;
                 }
             }
         }
     }
     return ;
 }

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 #define mt(a,b) memset(a,b,sizeof(a))
 using namespace std;
 const int M=;
 struct G{
     struct E{
         int u,v,next;
     }e[M<<];
     int le,head[M];
     void init(){
         le=;
         mt(head,-);
     }
     void add(int u,int v){
         e[le].u=u;
         e[le].v=v;
         e[le].next=head[u];
         head[u]=le++;
     }
 }g;
 int dp[M],num[M],n;
 void dfs(int u,int fa){
     dp[u]=;
     num[u]=;
     for(int i=g.head[u];~i;i=g.e[i].next){
         int v=g.e[i].v;
         if(v!=fa){
             dfs(v,u);
             num[u]+=num[v];
             dp[u]=max(dp[u],num[v]);
         }
     }
     dp[u]=max(dp[u],n-num[u]);
 }
 int main(){
     int t;
     while(~scanf("%d",&t)){
         while(t--){
             scanf("%d",&n);
             g.init();
             for(int i=,u,v;i<n-;i++){
                 scanf("%d%d",&u,&v);
                 g.add(u,v);
                 g.add(v,u);
             }
             dfs(,-);
             int sma=n;
             for(int i=;i<=n;i++){
                 sma=min(sma,dp[i]);
             }
             for(int i=;i<=n;i++){
                 if(dp[i]==sma){
                     printf("%d %d\n",i,dp[i]);
                     break;
                 }
             }
         }
     }
     return ;
 }

Godfather http://poj.org/problem?id=3107

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define mt(a,b) memset(a,b,sizeof(a))
 using namespace std;
 const int M=;
 struct G{
     struct E{
         int u,v,next;
     }e[M<<];
     int le,head[M];
     void init(){
         le=;
         mt(head,-);
     }
     void add(int u,int v){
         e[le].u=u;
         e[le].v=v;
         e[le].next=head[u];
         head[u]=le++;
     }
 }g;
 int dp[M],num[M],n;
 void dfs(int u,int fa){
     dp[u]=;
     num[u]=;
     for(int i=g.head[u];~i;i=g.e[i].next){
         int v=g.e[i].v;
         if(v!=fa){
             dfs(v,u);
             num[u]+=num[v];
             dp[u]=max(dp[u],num[v]);
         }
     }
     dp[u]=max(dp[u],n-num[u]);
 }
 int main(){
     while(~scanf("%d",&n)){
         g.init();
         for(int i=,u,v;i<n-;i++){
             scanf("%d%d",&u,&v);
             g.add(u,v);
             g.add(v,u);
         }
         dfs(,-);
         int sma=n;
         for(int i=;i<=n;i++){
             sma=min(sma,dp[i]);
         }
         for(int i=;i<=n;i++){
             if(dp[i]==sma){
                 printf("%d ",i);
             }
         }
         puts("");
     }
     return ;
 }

end