BZOJ4401 块的计数

题面:https://www.lydsy.com/JudgeOnline/problem.php?id=4401

显然没有无解的情况,因为可分成块内点的数量为1,由此拓展

设分成块内点的个数为i  等价于 得有 n/i个块 等价于 有n/i个点的【子树中点个数(size)】是i的倍数

另一个结论:一棵树划分为k份的方案是唯一的(为啥???)

#include<bits/stdc++.h>
using namespace std; 
const int maxn=1000004;
int sz[maxn],ans[maxn],tot=0,cnt=0,head[maxn],n;
struct node{
    int to,next;
}e[maxn<<1];
inline void add(int u,int v){
    e[cnt].to=v;e[cnt].next=head[u];head[u]=cnt++;
}
inline void dfs(int u,int fa){
    sz[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v!=fa){
            dfs(v,u);
            sz[u]+=sz[v];
        }
    }
    ans[sz[u]]++;
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int x,y;scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs(1,0);
    for(int i=1;i<=n;i++){
        int x=0;
        for(int j=i;j<=n;j+=i) x+=ans[j];
        if(x*i==n) tot++;
    }
    cout<<tot;
}

猜你喜欢

转载自www.cnblogs.com/wifimonster/p/10326693.html