洛谷 P1332 血色先锋队

第一次写bfs，纪念一下 qwq

之前一直都在写dfs，对bfs有莫名的恐惧

现在觉得。。。好像差不多 qwq

题目描述

巫妖王的天灾军团终于卷土重来，血色十字军组织了一支先锋军前往诺森德大陆对抗天灾军团，以及一切沾有亡灵气息的生物。孤立于联盟和部落的血色先锋军很快就遭到了天灾军团的重重包围，现在他们将主力只好聚集了起来，以抵抗天灾军团的围剿。可怕的是，他们之中有人感染上了亡灵瘟疫，如果不设法阻止瘟疫的扩散，很快就会遭到灭顶之灾。大领主阿比迪斯已经开始调查瘟疫的源头。原来是血色先锋军的内部出现了叛徒，这个叛徒已经投靠了天灾军团，想要将整个血色先锋军全部转化为天灾军团！无需惊讶，你就是那个叛徒。在你的行踪败露之前，要尽快完成巫妖王交给你的任务。

军团是一个N行M列的矩阵，每个单元是一个血色先锋军的成员。感染瘟疫的人，每过一个小时，就会向四周扩散瘟疫，直到所有人全部感染上瘟疫。你已经掌握了感染源的位置，任务是算出血色先锋军的领主们感染瘟疫的时间，并且将它报告给巫妖王，以便对血色先锋军进行一轮有针对性的围剿。

输入输出格式

输入格式：

第1行：四个整数N，M，A，B，表示军团矩阵有N行M列。有A个感染源，B为血色敢死队中领主的数量。

接下来A行：每行有两个整数x，y，表示感染源在第x行第y列。

接下来B行：每行有两个整数x，y，表示领主的位置在第x行第y列。

【数据规模】

1<=M,N<=500

1<=A,B<=M*N

输出格式：

第1至B行：每行一个整数，表示这个领主感染瘟疫的时间，输出顺序与输入顺序一致。如果某个人的位置在感染源，那么他感染瘟疫的时间为0。

输入输出样例

输入样例#1：复制

输出样例#1：复制

3
1
3

说明

如下图，标记出了所有人感染瘟疫的时间以及感染源和领主的位置。

思路：广搜，bfs预处理，然后O(1)查询

#include <algorithm>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
const int M = 505;
int n, m, a, b;
int p, q;
int map[M][M];
int dx[4] = {0, 1, 0, -1};
int dy[4] = {1, 0, -1, 0};
struct nond {
    int x, y;
    int t;
};

queue<nond> Q;
void bfs() {
    while (!Q.empty()) {
        nond now = Q.front();
        Q.pop();
        for (int i = 0; i < 4; ++i) {
            int cx = now.x + dx[i];
            int cy = now.y + dy[i];
            if (cx >= 1 && cx <= n && cy >= 1 && cy <= m && map[cx][cy] == 0) {
                map[cx][cy] = now.t + 1;
                Q.push((nond) {cx, cy, now.t + 1});
            }
        }
    }
}

int main() {
    scanf("%d%d%d%d", &n, &m, &a, &b);
    for (int i = 1; i <= a; ++i) {
        scanf("%d%d", &p, &q);
        Q.push((nond) {p, q, 0});
        map[p][q] = -1;
    }
    bfs();
    for (int i = 1; i <= b; ++i) {
        scanf("%d%d", &p, &q);
        if (map[p][q] == -1) printf("0\n");
        else printf("%d\n", map[p][q]);
    }
    return 0;
}