PAT 1076. Forwards on Weibo (BFS + 剪枝)

一开始想floyd, 但n^3 = 10^9, 要10秒了可能；

土一点的dijkstra超时, 因为dijkstra中寻找当前未访问过的距离最小值时，是对所有点进行遍历；以及对每个这样的点，是遍历map[1010][1010]数组。这样导致每遍dijkstra就是O(2*n*n)的复杂度。如果m >= n, 仍可能出现n^3的情况；

然后及用priority_queue来维护当前未访问过的距离最小值，用邻接表来维护每个点的相邻结点。

事实上，这就退化成BFS了.... 所以BFS+剪枝就能搞定的。(代码从135行变成了75行)

注意到，这里的边是有向边 —— "n l"后的第 i 行输入：

代表用户 i 关注了 m 个用户, 进而是该m个的用户的微博可以被用户 i 转发，从而有向边应该从那些 m 个用户指向 i.

代码。耗时2s. 有人耗时200+ms就完成了，知道怎么做的教我一下呗。

#include <iostream>#include <queue>#include <list>#include <cstring>using namespace std;struct Node{int m_id;int m_level;Node() {}Node(int id, int level): m_id(id), m_level(level) {}};int n, l, k, id;bool in_queue[1010];list<int> edge[1010];queue<Node> q;int bfs(Node now){//cout << "bfs" << endl;int cnt = -1;while (q.empty() == false){q.pop();}q.push(now);in_queue[now.m_id] = true;while (q.empty() == false){Node node = q.front();//in_queue[node.m_id] = true;q.pop();if (node.m_level > l){break;}//cout << node.m_id << endl;++ cnt;for (auto it = edge[node.m_id].begin(); it != edge[node.m_id].end(); ++ it){if (in_queue[*it] == false){q.push( Node(*it, node.m_level+1) );in_queue[*it] = true;}}}return cnt;}int main(){scanf("%d%d", &n, &l);for (int i = 1; i <= n; ++ i){scanf("%d", &k);for (int j = 0; j < k; ++ j){scanf("%d", &id);edge[id].push_back(i);}}scanf("%d", &k);for (int i = 0; i < k; ++ i){memset(in_queue, 0, sizeof(in_queue));scanf("%d", &id);printf("%d\n", bfs(Node(id, 0)));}return 0;}

耗时220ms.. 不懂怎么做到的