poj 3107(图经典的DFS+回溯)

来源:互联网 发布:网络osi的七层模型 编辑:程序博客网 时间:2024/06/07 00:14

    给出一个树,求删除一个点后,他的子树最大节点数的最小值。有点绕啊,,,~~

    从第一个点进行DFS,求出每个节点除了递归的上层之外的叶子的子树最大节点数m,递归的过程中求出联通本身的节点数k,则该节点的最大子树就是max(m,n-k);有点像dp.

    dfs还要多练~~~~~~~~~~~~~~~~~~~

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
const int MAX=50005;
using namespace std;
int top=MAX;
int n;


struct Node
{
    int v;
    int next;
}mp[MAX*2+1];
int head[MAX];
int cnt;


void init()
{
     cnt=0;
     for(int i=1;i<=n;i++)
     {
         head[i]=-1;
     }
}


void insert(int a,int b)
{
   mp[cnt].v=b;
   mp[cnt].next=head[a];
   head[a]=cnt;
   cnt++;
   mp[cnt].v=a;
   mp[cnt].next=head[b];
   head[b]=cnt;
   cnt++;
}


int vis[MAX];
int num[MAX];
int ans[MAX];
int k=0;


void DFS(int u)
{
    int mi=-1;
    vis[u]=1;
    for(int i=head[u];i!=-1;i=mp[i].next)
    {
        int to=mp[i].v;
        if(!vis[to])
        {
            DFS(to);
            num[u]+=num[to];//求除了递归上层之外的所有节点数
            mi=max(mi,num[to]);//u点的最大子树
        }
    }
    mi=max(mi,n-num[u]);
    if(mi<top)
    {
        ans[0]=u;
        k=1;
        top=mi;
    }
    else if(mi==top)
    {
        ans[k++]=u;
    }
}


int main()
{
   scanf("%d",&n);
   init();
   int u,v;
   for(int i=1;i<n;i++)
   {
       scanf("%d%d",&u,&v);
       insert(u,v);
   }


   for(int i=1;i<=n;i++)
   num[i]=1;


   memset(vis,0,sizeof(vis));
   k=0;
   top=MAX;
   DFS(1);
   sort(ans,ans+k);


   for(int i=0;i<k;i++)
   cout<<ans[i]<<' ';
   return 0;
}

原创粉丝点击