1004 Counting Leaves

最新推荐文章于 2024-10-16 18:29:10 发布

QZZ_PP

最新推荐文章于 2024-10-16 18:29:10 发布

阅读量62

点赞数

文章标签：数据结构

本文链接：https://blog.csdn.net/QZZ_PP/article/details/130709211

版权

该文提供了两个C++程序，分别实现了BFS（广度优先搜索）和DFS（深度优先搜索）算法，用于程序设计类实验辅助教学。在给定的图结构中，BFS程序计算从顶点1开始的每一层的节点数，而DFS程序则统计以顶点1为起点的每个深度的节点数量。

摘要由CSDN通过智能技术生成

PTA | 程序设计类实验辅助教学平台

bfs层序遍历：

#include<bits/stdc++.h>
using namespace std;
const int N=110;
int g[N][N];
int n,m,id,x,k;
vector<int>v;
int has_son[N];
void bfs(int u)
{
    queue<int>q;
    q.push(u);
    while(q.size())
    {
        int flag=0;
        int s=q.size();
        while(s--)
        {
            int t=q.front();
            q.pop();
            for(int i=1;i<=n;i++)
            {
                if(g[t][i])
                {
                    has_son[t]=1;
                    q.push(i);
                }
            }
            if(has_son[t]==0)
            {
                flag++;
            }
        }
        v.push_back(flag);
    }
}
int main()
{
    scanf("%d %d",&n,&m);
    while(m--)
    {
        scanf("%d %d",&id,&k);
        for(int i=0;i<k;i++)
        {
            int x;
            scanf("%d",&x);
            g[id][x]=1;
        }
    }
    bfs(1);
    printf("%d",v[0]);
    for(int i=1;i<v.size();i++)
    {
        printf(" %d",v[i]);
    }
    return 0;
}

dfs：

#include<bits/stdc++.h>
using namespace std;
const int N=110;
int e[N],h[N],ne[N],idx;
int max_depth;
int ans[N];
int n,m;
void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int depth)
{
    if(h[u]==-1)
    {
        ans[depth]++;
        max_depth=max(max_depth,depth);
        return;
    }
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        dfs(j,depth+1);
    }
}
int main()
{
    memset(h,-1,sizeof h);
    scanf("%d %d",&n,&m);
    while(m--)
    {
        int id,k,son;
        scanf("%d %d",&id,&k);
        while(k--)
        {
            scanf("%d",&son);
            add(id,son);
        }
    }
    dfs(1,0);
    printf("%d",ans[0]);
    for(int i=1;i<=max_depth;i++)
    {
        printf(" %d",ans[i]);
    }
}