【bzoj4401】块的计数（水dfs）

　　题目传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=4401

　　假设把树划分为x个节点作一块，那么显然只有当x|n的时候才可能存在划分方案，并且这种划分方案是唯一的。

　　并且对于一棵树，只有当有n/x个节点的子树大小%x==0的时候才可能存在划分方案，因为如果把一棵树的根节点及其所在的块切掉，那么剩下的子树若存在划分方案，一定满足这些子树的节点个数%x==0。

　　所以这道题就变成一道水题了。

　　代码（我本来想着用bfs代替dfs会跑得快一点，然而似乎并没有什么卵用）：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
#define ll long long
#define min(a,b) (a<b?a:b)
#define max(a,b) (a>b?a:b)
ll read()
{
    ll tmp=; char f=,c=getchar();
    while(c<''||''<c){if(c=='-')f=-; c=getchar();}
    while(''<=c&&c<=''){tmp=tmp*+c-''; c=getchar();}
    return tmp*f;
}
using namespace std;
int size[],q[],fa[],cnt[];
int fir[],ne[],to[];
int n,tot=;
void add(int x,int y){to[++tot]=y; ne[tot]=fir[x]; fir[x]=tot;}
int main()
{
    int i;
    n=read();
    for(i=;i<n;i++){
        int x=read(),y=read();
        add(x,y); add(y,x);
    }
    int h=,t=; q[]=; fa[]=-;
    while(h<=t){
        for(i=fir[q[h]];i;i=ne[i])
            if(!fa[to[i]])q[++t]=to[i],fa[to[i]]=q[h];
        ++h;
    }
    for(i=n;i;i--){
        size[q[i]]=;
        for(int j=fir[q[i]];j;j=ne[j])
            if(to[j]!=fa[q[i]])size[q[i]]+=size[to[j]];
        ++cnt[size[q[i]]];
    }
    int ans=;
    for(i=;i<=n;i++)
        if(n%i==){
            int tmp=;
            for(int j=i;j<=n;j+=i)tmp+=cnt[j];
            if(tmp*i==n)++ans;
        }
    printf("%d\n",ans);
    return ;
}

跑得慢也会输