1076 Forwards on Weibo (链接表层序遍历)

题意:给出关注列表,博主的粉丝会给博主点赞,粉丝的粉丝也会给博主点赞,一直递推到最多L层,求,最后会有多少人给博主点赞。

思路:将关注的粉丝用链接表存储,再对博主进行层序遍历,遍历L+1层(因为不能包含博主层),并且将遍历过的人都标记防止重复计算,同时算出所有遍历到的所有结点。结点数-1(不包含博主)即为答案。

(刚开始写了个深度为L+1的深度优先遍历,结果不对,因为深度遍历过的结点可能会与后面的兄弟结点重复,造成提前标记,故只能使用层序遍历)

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;

vector<int>son[1010];
bool vis[1010];
int bfs(int u,int k){
    memset(vis,0,sizeof vis);
    queue<int>q;
    q.push(u);
    k++;
    int res=0;
    while(q.size()&&k--){
        int len=q.size();
        while(len--){
            int f=q.front();
            q.pop();
            if(vis[f])continue;//防止重复计算
            res++;
            vis[f]=1;
            for(auto x:son[f]){
                if(!vis[x]){
                    q.push(x);
                }
            }
        }
    }
    return res;
}
int main(){
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        int m;cin>>m;
        while(m--){
            int a;cin>>a;
            son[a].push_back(i);
        }
    }
    int q;cin>>q;
    while(q--){
        int a;cin>>a;
        cout<<bfs(a,k)-1<<endl;
    }
}
相关推荐
卷福同学1 小时前
【养虾日记】Openclaw操作浏览器自动化发文
人工智能·后端·算法
春日见2 小时前
如何入门端到端自动驾驶?
linux·人工智能·算法·机器学习·自动驾驶
图图的点云库2 小时前
高斯滤波实现算法
c++·算法·最小二乘法
一叶落4383 小时前
题目:15. 三数之和
c语言·数据结构·算法·leetcode
y = xⁿ3 小时前
【LeetCodehot100】2:两数相加 19 删除链表倒数第n个节点
数据结构·链表
老鱼说AI4 小时前
CUDA架构与高性能程序设计:异构数据并行计算
开发语言·c++·人工智能·算法·架构·cuda
罗湖老棍子4 小时前
【例 1】数列操作(信息学奥赛一本通- P1535)
数据结构·算法·树状数组·单点修改 区间查询
big_rabbit05024 小时前
[算法][力扣222]完全二叉树的节点个数
数据结构·算法·leetcode
张李浩5 小时前
Leetcode 15三题之和
算法·leetcode·职场和发展
2301_793804696 小时前
C++中的适配器模式变体
开发语言·c++·算法