树上dfs+思维
1 #include<cstdio> 2 const int N=1000010; 3 int cnt,head[N],n; 4 int size[N],num[N]; 5 void init() 6 { 7 cnt = 0; 8 for(int i =0;i<N;i++) head[i]=-1; 9 } 10 struct Node{ 11 int fr,to,nex; 12 }e[N<<1];//乘2 13 void add(int x,int y){ 14 e[cnt].fr=x;e[cnt].to=y; 15 e[cnt].nex=head[x];head[x]=cnt++; 16 } 17 void dfs(int x,int rt){ 18 size[x] =1; 19 for(int i =head[x];i!=-1;i=e[i].nex){ 20 int v= e[i].to; 21 if(v!=rt){ 22 dfs(v,x); 23 size[x]+=size[v]; 24 } 25 } 26 num[size[x]]++; 27 } 28 int x,y; 29 bool check(int i) 30 { 31 i++;//子树数目 32 if(n%i) return false;//无法均分 33 int tmp = n/i;//每个子树大小 34 int sum=0; 35 for(int i=tmp;i<=n;i+=tmp) sum+=num[i];//找子树大小为tmp的子树数目 36 //tmp 2*tmp 3*tmp 其实都是再找tmp 37 return sum==i; 38 } 39 int main() 40 { 41 //freopen("1.txt","r",stdin); 42 init(); 43 scanf("%d",&n); 44 for(int i =0;i<n-1;i++) { 45 scanf("%d%d",&x,&y); 46 add(x,y);add(y,x);//无向边 47 } 48 dfs(1,0);//1不一定是根节点dfs(n-1,0)都可以。 49 for(int i=1;i<n;i++){ 50 //printf("%d......",i); 51 if(check(i)) printf("%d ",i); 52 } 53 return 0; 54 }
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现