PAT Advanced Level 1076. Forwards on Weibo (30)

【来源】

1076. Forwards on Weibo (30)

【分析】

本题给出微博用户的关注情况，在假定某一用户的粉丝看到某条微博一定会转发的情况下，求可能被转发的次数。

抽象一下就是一个限定了层数的广度优先搜索（BFS）。

基本的数据结构是以vector为元素类型的一个vector。该vector的index和用户id相关，index所对应的元素为此id的粉丝们。

给定发微博的用户以后，先设定除博主之外的所有用户都没有转发。然后用一个队列来进行BFS，出队时将此用户的转发设置为True。遍历完毕之后统计标记为True的用户的数量，输出即可。

为了统计搜索的层数，每次搜索之前我们把队列中的最后一个元素记录下来，为endofLevel 变量。每当我们发现出队的元素为endofLevel时，表明一层遍历的结束，然后更新endofLevel的值并将层数加1。当层数达到所限定的层数L时，停止搜索即可。

【代码】

#include <iostream>#include <cstdio>#include <vector>#include <deque>using namespace std;int main(){    int N, L;    scanf("%d%d", &N, &L);    vector<vector<int>> fans(N);    for (int i = 0; i < N; ++i){        int num;        scanf("%d", &num);        for (int j = 0; j < num; ++j){            int followee;            scanf("%d", &followee);            fans[followee-1].push_back(i);        }    }        int K;    scanf("%d", &K);        for (int i = 0; i < K; ++i){        bool* hasfoward = new bool[N];        for (int i = 0; i < N; ++i){            hasfoward[i] = false;        }        int blogger;        scanf("%d", &blogger);        hasfoward[blogger - 1] = true;        deque<int> queue;        queue.push_back(blogger - 1);                int currentlevel = 0;        int last = blogger - 1;        int endOfLevel = queue.back();        while (!queue.empty()){             int front = queue.front();            queue.pop_front();              hasfoward[front] = true;            for (int j = 0; j < fans[front].size(); ++j){                if (!hasfoward[fans[front][j]]){                    queue.push_back(fans[front][j]);                    hasfoward[fans[front][j]] = true;                                    }            }            if (front == endOfLevel)            {                endOfLevel = queue.back();                ++currentlevel;            }            if (currentlevel >= L){                break;            }                    }        int fowards = 0;        for (int i = 0; i < N; ++i){            if (hasfoward[i]){                ++fowards;            }        }        printf("%d\n", fowards-1);        delete[] hasfoward;    }        system("pause");    return 0;}

【点评】

此题考察BFS的应用，以及BFS层数计数问题。注意记录下某一用户的粉丝，而不是像题目给出的那样记录关注的人，这样搜索起来会很方便。