题目链接https://pintia.cn/problem-sets/994805342720868352/problems/994805392092020736
开始用dijkstra,把follow距离当作1,结果超时了。。。那么就是图的BFS问题,最深层数为L
。然而忘记BFS怎么写了。。自己写了一个,最后一个点一直超时,然后照着柳神的抄了一遍(BFS部分),提交发现还是最后一个点超时,于是把每个人的fans的存储从set改成了vector,然后就过了。。。
这篇就写全一点好了
读取输入,注意给的是每个人followings的列表,我们要转换为每个人的fans列表。
for (int i = 1; i <= N; i++) {
int M;
scanf("%d", &M);
for (int j = 0; j < M; j++) {
int tmp;
scanf("%d", &tmp);
cf[tmp].push_back(i);
}
}
存储的类,包含id
和层数lvl
class Node {
public:
int id, lvl;
Node() {}
Node(int id, int lvl) {
this->id = id;
this->lvl = lvl;
}
};
BFS部分
创建队列,把第一个Nodet
压进队列,标记inq[t]=true
,表示它已经进过队列了,然后在队列为空之前循环:
bool inq[1002] = { false };
queue<Node> q;
q.push(t);
inq[t.id] = true;
int cnt = 0;
将队列最前部的Nodetop
取出,如果层数小于L
就继续,否则就要停下来了。
对top
的每一个fan,如果这个fan没有进过队列,就将它的id取出,并将lvl置为top
的层数+1,组成新Node,放进队列,计数cnt++
while (!q.empty()) {
Node top = q.front();
q.pop();
int top_id = top.id;
if (top.lvl < L){
for (int i = 0; i < cf[top_id].size(); i++) {
int next_id = cf[top_id][i];
if (!inq[next_id]){
Node nxt(next_id, top.lvl + 1);
q.push(nxt);
inq[nxt.id] = true;
cnt++;
}
}
}
}
最后把cnt
返回,BFS就结束了
输出,用一个初始都为-1
的数组ret
存BFS的结果。这样的话,如果每次query读入的是相同的节点,就不需要重新bfs一遍了。
scanf("%d", &K);
for (int i = 0; i < K; i++) {
int s;
scanf("%d", &s);
if (ret[s] < 0){
Node t(s, 0);
int tmp_ret = bfs(t);
ret[s] = tmp_ret;
}
printf("%d\n", ret[s]);
}
完整代码
#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<map>
#include<set>
#include<queue>
#include<string.h>
using namespace std;
int N, L, K;
vector<vector<int>> cf;
class Node {
public:
int id, lvl;
Node() {}
Node(int id, int lvl) {
this->id = id;
this->lvl = lvl;
}
};
int bfs(Node t) {
bool inq[1002] = { false };
queue<Node> q;
q.push(t);
inq[t.id] = true;
int cnt = 0;
while (!q.empty()) {
Node top = q.front();
q.pop();
int top_id = top.id;
if (top.lvl < L){
for (int i = 0; i < cf[top_id].size(); i++) {
int next_id = cf[top_id][i];
if (!inq[next_id]){
Node nxt(next_id, top.lvl + 1);
q.push(nxt);
inq[nxt.id] = true;
cnt++;
}
}
}
}
return cnt;
}
int main() {
scanf("%d %d", &N, &L);
cf.resize(N + 1);
vector<int> ret(N+1, -1);
for (int i = 1; i <= N; i++) {
int M;
scanf("%d", &M);
for (int j = 0; j < M; j++) {
int tmp;
scanf("%d", &tmp);
cf[tmp].push_back(i);
}
}
scanf("%d", &K);
for (int i = 0; i < K; i++) {
int s;
scanf("%d", &s);
if (ret[s] < 0){
Node t(s, 0);
int tmp_ret = bfs(t);
ret[s] = tmp_ret;
}
printf("%d\n", ret[s]);
}
return 0;
}