【DFS+思维】 GYM101620J Justified Jungle

最新推荐文章于 2021-05-28 21:37:14 发布

原创最新推荐文章于 2021-05-28 21:37:14 发布 · 243 阅读

0 ·

CC 4.0 BY-SA版权

思维题同时被 2 个专栏收录

17 篇文章

订阅专栏

dfs

12 篇文章

订阅专栏

本文深入探讨了一种基于图论的算法实现，详细介绍了如何通过深度优先搜索（DFS）和图的遍历技巧来解决特定问题。文章展示了如何构建图的数据结构，包括节点、边和邻接表，并解释了如何利用这些结构进行有效的路径查找和节点访问。此外，还讨论了算法的时间复杂度和优化策略。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://codeforces.com/gym/101620/attachments

#include <bits/stdc++.h>
using namespace std;
set <int> s;
int head[1000005],to[2000005],nxt[2000005],Size[1000005],num[1000005];
int tot,n;

void addedge(int u,int v)
{
    to[++tot]=v;
    nxt[tot]=head[u];
    head[u]=tot;
}

void dfs(int u,int fa)
{
    Size[u]=1;
    for(int i=head[u];i!=-1;i=nxt[i])
    {
        int v=to[i];
        if(v==fa) continue;
        dfs(v,u);
        Size[u]+=Size[v];
    }
    num[Size[u]]++;
}

int ok(int x)
{
    int ans=0;
    for(int i=x;i<=n;i+=x)
    {
        ans+=num[i];
    }
    return ans;
}

int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    tot=0;
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        addedge(u,v);
        addedge(v,u);
    }
    dfs(1,0);
   // int k=sqrt(n);
    for(int i=1;i*i<=n;i++)
    {
        if(n%i==0)
        {
            int q=ok(i);
            if(q==n/i)
            {
                s.insert(q-1);
            }
            if(i*i!=n&&i!=1)
            {
                q=ok(n/i);
                if(q==i) s.insert(q-1);
            }
        }
    }
    for(set<int>::iterator it = s.begin(); it != s.end(); ++it)
    {
        int p = *it;
        printf("%d ",p);
    }
    printf("\n");
    return 0;
}