BZOJ 4401 块的计数 [DFS序]

传送门

因为块的大小那么方案最多只有1种，
发现$O(n\sqrt n)$乱搞，不过会TLE。。？
接着我们又发现，一个节点可以作一个块的根，当且仅当该节点的size能被块的大小整除。。。。然后就可以$O(nlog^n)$搞了..
大水题，块状树在哪？

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<set>
#include<queue>
#include<algorithm>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<stack>
#define rez(i,x,y) for(int i=x;i>=y;i--)
#define res(i,x,y) for(int i=x;i<=y;i++)
#define INF 2100000000
#define ll long long
#define clr(x)  memset(x,0,sizeof(x))

using namespace std;
inline void read(int &res){
    static char ch;register int flag=1;
    while((ch=getchar())<'0'||ch>'9')if(ch=='-')flag=-1;res=ch-48;
    while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
const int N = 1000005;
vector<int>E[N];
int sz[N],cnt[N],ans,n;
void dfs(int u,int f){
    sz[u]=1;
    for(register int v,i=0;i<E[u].size();i++)
        if(v=E[u][i],v!=f)dfs(v,u),sz[u]+=sz[v];
    cnt[sz[u]]++;
}
int main(){
    read(n);
    for(register int u,v,i=1;i<n;i++)read(u),read(v),E[u].push_back(v),E[v].push_back(u);
    dfs(1,1);
    for(register int i=1;i<=n;i++){
        for(register int j=2;j*i<=n;j++)
            cnt[i]+=cnt[i*j];
        if(cnt[i]*i==n)ans++;
    }
    printf("%d\n",ans);
    return 0;
}

vector 居然稳过。。