bzoj4401: 块的计数

C++图论算法解析

蜗角虚名羊

于 2019-02-26 14:41:52 发布

阅读量187

点赞数

分类专栏：思维

思维专栏收录该内容

39 篇文章

订阅专栏

本文深入探讨了使用C++实现的一种图论算法，详细介绍了如何通过深度优先搜索（DFS）来解决特定的图问题。文章中包含了一个完整的C++代码示例，展示了如何构建图的邻接表、进行深度优先遍历以及计算特定节点的子树大小。通过对算法的逐步解析，读者可以更好地理解图论中节点与边的关系，以及如何利用C++高效地处理图结构。

题目
 题解

Code

#include<bits/stdc++.h>
using namespace std;
const int N=1000002;
struct node{
	int to,ne;
}e[N<<1];
int sz[N],x,y,tot,h[N],num[N],n,i,cnt,ans,j;
inline char gc(){
	static char buf[100000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd(){
	int x=0,fl=1;char ch=gc();
	for (;ch<48||ch>57;ch=gc())if(ch=='-')fl=-1;
	for (;48<=ch&&ch<=57;ch=gc())x=(x<<3)+(x<<1)+(ch^48);
	return x*fl;
}
inline void wri(int a){if(a<0)a=-a,putchar('-');if(a>=10)wri(a/10);putchar(a%10|48);}
inline void wln(int a){wri(a);puts("");}
void add(int x,int y){e[++tot]=(node){y,h[x]},h[x]=tot;}
void dfs(int u,int fa){
	sz[u]=1;
	for (int i=h[u],v;i;i=e[i].ne)
		if ((v=e[i].to)!=fa) dfs(v,u),sz[u]+=sz[v];
}
int main(){
	n=rd();
	for (i=1;i<n;i++) x=rd(),y=rd(),add(x,y),add(y,x);
	dfs(1,0);
	for (i=1;i<=n;i++) num[sz[i]]++;
	for (i=1;i<=n;i++)
		if (n%i==0){
			cnt=0;
			for (j=1;j<=n/i;j++) cnt+=num[i*j];
			ans+=cnt==n/i;
		}
	printf("%d",ans);
}