【jzoj4757】【树上摩托】

题目大意

把这棵树砍去任意条边，拆成若干棵新树，让每棵新树的节点数相同。

有多少种方法分割这棵树。

解题思路

首先可以变成有根树，枚举每棵树的节点数k，则size为k的倍数的个数有等于n/k。我们可以开一个桶存size，枚举size统计个数判断即可。

暴力code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LF double
#define LL long long
#define max(a,b) ((a>b)?a:b)
#define min(a,b) ((a>b)?b:a)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=1000000,maxm=100,inf=2147483647;
int n,m,gra,size[maxn+10],du[maxn+10],q[maxn+10],begin[maxn+10],to[maxn*2+10],next[maxn*2+10];
bool visit[maxn+10];
int read(){
    char ch=getchar();int val=0;
    for(;(ch<'0')||(ch>'9');ch=getchar());
    for(;(ch>='0')&&(ch<='9');val=val*10+ch-'0',ch=getchar());
    return val;
}
void insert(int u,int v){
    to[++gra]=v;
    next[gra]=begin[u];
    begin[u]=gra;
}
void dfs(int now){
    int l=0,r=0;q[++r]=now;
    visit[now]=1;
    for(;l!=r;){
        now=q[++l];
        for(int i=begin[now];i;i=next[i])
            if(!visit[to[i]]){
                q[++r]=to[i];
                visit[to[i]]=1;
            }
    }
    fd(j,r,1){
        size[q[j]]=1;now=q[j];
        for(int i=begin[now];i;i=next[i])
            if(!visit[to[i]])
                size[now]+=size[to[i]];
        visit[q[j]]=0;
    }
}
int main(){
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    n=read();
    fo(i,1,n-1){
        int u=read(),v=read();du[u]++;du[v]++;
        insert(u,v);
        insert(v,u);
    }
    fo(i,1,n)
        if(du[i]==1){
            dfs(i);
            break;
        }
    int ans=0,sqrtn=sqrt(n),cnt,ii;
    fo(i,1,sqrtn)
        if(n%i==0){
            cnt=0;
            fo(j,1,n)
                if(size[j]%i==0)cnt++;
            if(cnt==n/i)
                ans++;
            cnt=0;ii=n/i;
            if(i==ii)continue;
            fo(j,1,n)
                if(size[j]%ii==0)cnt++;
            if(cnt==i)
                ans++;
        }
    printf("%d",ans);
    return 0;
}