POJ - 2378 Tree Cutting(递归回溯的应用---一个子节点如何判断是否符合条件)

题目链接:https://cn.vjudge.net/contest/309172#problem/Z
Sample Input

10
1 2
2 3
3 4
4 5
6 7
7 8
8 9
9 10
3 8

Sample Output

3
8

解析:输入N个点。接下来N-1个关系。判断去除某个点，分成的部分。每一部分点的个数如果都 <=N/2 ,则这个点符合情况。最后从小到大输出这些点。
注:(N表示点的个数，M表示边数)，对于M<N*N的稀疏图，用邻接表存，对于M较大的，用邻接矩阵存。

对于上图:如果去除1号顶点。将上图分成了3部分，两部分为1的子边，个数分别为1和3（这里1和3不包括本身）。递归下去，代码中k为子边的点数，ans为所有子边点数的和。需要判断所有的子部分是剖满足k<=N/2,还有N-asn是否<=N/2。

代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int u[20001],v[20001],first[20001],next[20001];
int book[20001],a[20001];
int N,m,k;
void serve(int a,int b){
    u[k]=a,v[k]=b;
    next[k]=first[a];//无向图邻接表的存法
    first[a]=k;
    k++;
}
int dfs(int x){
    int ans=1,flag=0;
    for(int i=first[x]; i!=-1; i=next[i]){  //以x为顶点的所有边
        if(book[v[i]]==0){
            book[v[i]]=1;
            int k=dfs(v[i]);//k为以这个点为父亲，一条路径上子节点的个数
            if(k>N/2)//每一个子部分是否符合情况
                flag=1;
            ans+=k;
        }
    }
    if(flag==0&&N-ans<=N/2)
        a[m++]=x;
    return ans;
}
int main(){
    while(~scanf("%d",&N)){
        k=1,m=0;
        memset(book,0,sizeof(book));
        memset(first,-1,sizeof(first));
        for(int i=1; i<N; i++){
            int a,b;
            scanf("%d%d",&a,&b);
            serve(a,b);
            serve(b,a);
        }
        book[1]=1;
        dfs(1);
        sort(a,a+m);
        if(m==0)
            printf("NONE\n");
        else{
            for(int i=0; i<m; i++)
                printf("%d\n",a[i]);
        }
    }
    return 0;
}