2017-2018 ACM-ICPC, Central Europe Regional Contest (CERC 17) J - Justified Jungle（dfs+思维）

题目链接：http://codeforces.com/gym/101620/attachments

题意：给一棵树，然后删除几条边，使得所有子树的节点个数相同，输出可以删除的边的个数。

思路：如果要删除k条边，就会分出k+1个子树，每个子树的节点数为n/(k+1)，所以k+1是n的因子。如果某个节点的子树大小为 n/(k+1)的倍数，那么这个节点与它的父节点连接的边就要删除（保证每一个连通块的大小都为n/(k+1)）。所以我们可以统计子树大小为 n/(k+1) 倍数的个数，如果个数等于 k+1，则说明可以删除k条边对树进行划分，否则就不能。可以首先建树，计算出所有节点的子树大小，然后遍历k，进行判断。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1000005;
int head[maxn]={0},to[maxn*2],next1[maxn*2],size[maxn];
int k,n;
int vis[maxn]={0},ans[maxn];
void addedge(int a,int b){//建树
    to[++k]=b;
    next1[k]=head[a];
    head[a]=k;
}
void dfs(int x,int y){//计算每个节点的子树大小
    size[x]=1;
    for(int i=head[x];i!=0;i=next1[i]){
        if(to[i]!=y){
            dfs(to[i],x);
            size[x]+=size[to[i]];
        }
    }
    vis[size[x]]++;//记录子树大小为size[x]的有多少个
}
int check(int x){
    int s=0;
    for(int i=x;i<=n;i+=x){
        s+=vis[i];
    }
    if(s==n/x) return 1;
    return 0;
}
int main(){
    ios::sync_with_stdio(false);
    int a,b;
    int t=0;
    cin>>n;
    k=0;
    for(int i=0;i<n-1;i++){
        cin>>a>>b;
        addedge(a,b);
        addedge(b,a);
    }
    dfs(1,0);
    for(int i=1;i*i<=n;i++){
        if(n%i==0){
            if(check(i))ans[t++]=n/i-1;
            if(check(n/i))ans[t++]=i-1;
        }
    }
    sort(ans,ans+t);
    for(int i=1;i<t;i++){
        cout<<ans[i]<<" ";
    }
    return 0;
}