BZOJ 3391 [Usaco2004 Dec]Tree Cutting网络破坏：dfs【无根树 节点分枝子树大小】

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=3391

题意：

　　给你一棵无根树，求分支size均不大于一半点数的点。

题解：

　　假定1为根。

　　dfs时统计siz[i]和par[i]。

　　对于每个节点判断一下子树大小siz[son]和自己往上的子树大小n - siz[now]。

AC Code:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <vector>
#define MAX_N 10005

using namespace std;

int n;
int par[MAX_N];
int siz[MAX_N];
vector<int> ans;
vector<int> edge[MAX_N];

void read()
{
    cin>>n;
    int a,b;
    for(int i=0;i<n-1;i++)
    {
        cin>>a>>b;
        edge[a].push_back(b);
        edge[b].push_back(a);
    }
}

void dfs(int now,int p)
{
    par[now]=p;
    siz[now]=1;
    for(int i=0;i<edge[now].size();i++)
    {
        int temp=edge[now][i];
        if(temp!=p)
        {
            dfs(temp,now);
            siz[now]+=siz[temp];
        }
    }
}

void solve()
{
    dfs(1,-1);
    for(int i=1;i<=n;i++)
    {
        int flag=true;
        for(int j=0;j<edge[i].size();j++)
        {
            int temp=edge[i][j];
            if(temp!=par[i] && siz[temp]*2>n)
            {
                flag=false;
                break;
            }
        }
        if((n-siz[i])*2>n) flag=false;
        if(flag) ans.push_back(i);
    }
}

void print()
{
    if(!ans.size()) cout<<"NONE"<<endl;
    else
    {
        for(int i=0;i<ans.size();i++)
        {
            cout<<ans[i]<<endl;
        }
    }
}

int main()
{
    read();
    solve();
    print();
}