开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 25 天,点击查看活动详情
分析
这题开始我就感觉比较离谱也比较划水,所以开始没看出来这是个bfs,tnnd,后来发现就是链式前向星建一下图,来具体统计一下哪些人有哪些粉丝,然后跑L次bfs就行了(在每次开始把队列内部的元素个数加上,然后再寻找一个这L重里面有没有其他间接的粉丝,有的话就入队,继续扩展。)最后一定记好了要+最后一次队列的元素个数,并且-1,减去开始本身,还是比较水的一题,但是一定要记好把数组开大,QAQ,数组小了就寄了,因为100组可能有重复的,所以要开到100010的空间
代码
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <map>
#include <set>
#include <iomanip>
#include <cmath>
#include <unordered_map>
#include <stack>
#include <queue>
#define ll long long
#define lowbit(x) x&(-x)
using namespace std;
typedef pair<int,int> PII;
typedef pair<string,int> PSI;
typedef stack<int> stk;
int gcd(int x,int y){return y?gcd(y,x%y):x;}
ll qmi(ll x,ll y,int mod){
ll res=1;
while(y){if(y&1) res=res*x%mod;y>>=1;x=x*x%mod;}
return res;
}
const int N=500010;
int n,L;
int idx=0,ne[N],e[N],h[N],k;
inline void add(int a,int b){
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
bool st[N];
int bfs(int x){
int res=0;
memset(st,false,sizeof st);
queue<int> q;
q.push(x);
st[x]=true;
for(int d=1;d<=L;d++){
int si=q.size();
res+=q.size();
for(int i=1;i<=si;i++){
int t=q.front();
q.pop();
for(int i=h[t];~i;i=ne[i]){
int j=e[i];
if(!st[j]){
st[j]=true;
q.push(j);
}
}
}
}
return res+q.size()-1;
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0),cout.tie(0);
cin>>n>>L;
memset(h,-1,sizeof h);
for(int i=1;i<=n;i++){
int num;
cin>>num;
while(num--){
int id;
cin>>id;
add(id,i);
}
}
cin>>k;
while(k--){
int x;
cin>>x;
cout<<bfs(x)<<"\n";
}
return 0;
}
希望能帮助到大家,QAQ!