1076. Forwards on Weibo (30) PAT 甲级

最新推荐文章于 2020-04-22 16:52:05 发布

原创最新推荐文章于 2020-04-22 16:52:05 发布 · 467 阅读

0 ·

CC 4.0 BY-SA版权

PAT 甲级专栏收录该内容

118 篇文章

订阅专栏

本文介绍了一种基于广度优先搜索(BFS)的图遍历算法实现，该算法能够在给定节点数量和层级限制的情况下，计算出从指定起点出发能够到达的节点数。通过使用队列和结构体来组织数据，有效地实现了对图结构的遍历。

传送门

#include<stdio.h>
#include<vector>
#include<queue>
#include<string.h>

#define MAX_V 1100

using namespace std;

struct Node{
    int v;
    int level;
}user;

int n,l;
vector<Node> G[MAX_V];
bool visited[MAX_V];


void bfs(int start){
    visited[start]=true;
    int fowardNum=0;
    queue<Node> q;
    Node s;
    s.v=start;
    s.level=0;
    q.push(s);

    while(!q.empty()){
        Node p=q.front();
        q.pop();
        int u=p.v;
        for(int i=0;i<G[u].size();i++){
            Node next=G[u][i];
            next.level=p.level+1;
            if(!visited[next.v]&&next.level<=l){
                q.push(next);
                visited[next.v]=true;
                fowardNum++;
            }
        }
    }
    printf("%d\n",fowardNum);
}

int main(){
    int followNum;
    int follow;
    int queryNum;
    int start;
    scanf("%d%d",&n,&l);
    for(int i=1;i<=n;i++){
        user.v=i;
        scanf("%d",&followNum);
        for(int j=0;j<followNum;j++){
            scanf("%d",&follow);
            G[follow].push_back(user);
        }
    }
    scanf("%d",&queryNum);
    for(int i=0;i<queryNum;i++){
        memset(visited,false,sizeof(visited));
        scanf("%d",&start);
        bfs(start);
    }
}