[POJ3107]Godfather（树形dp）

最新推荐文章于 2023-09-26 23:36:32 发布

原创最新推荐文章于 2023-09-26 23:36:32 发布 · 665 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#POJ #dp

收录于

题解同时被 2 个专栏收录

1097 篇文章

订阅专栏

245 篇文章

订阅专栏

本文详细介绍了如何解决POJ1655问题，包括题目解析、算法设计、代码实现及优化策略。通过实例分析，读者可以深入理解并掌握解决此类问题的方法。

题目描述

传送门
题意：同POJ1655，可能有多种方案，按从小到大输出。

题解

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int max_n=5e4+5;
const int max_e=max_n*2;
const int INF=2e9;
int T,n,ans=INF,x,y,num;
int tot,point[max_n],next[max_e],v[max_e],size[max_n],ansp[max_n];
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[num=1]=ansnowp;
    else if (ansnow==ans) ansp[++num]=ansnowp;
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<n;++i) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dfs(1,0);
    sort(ansp+1,ansp+num+1);
    for (int i=1;i<=num;++i) printf("%d%c",ansp[i]," \n"[i==num]);
}