普及练习场 广度优先搜索 01迷宫

题目链接

题意理解

这题如果每次查询再去搜索能到多少个格子肯定是不行的，需要预处理一下，这是一个点。另外，要能想到，如果一个点能到另外一个点，那么这两个点最后能到的格子数是一样多的，这是第二个点。可能第二个点会想着去用并查集做，但是其实不需要，直接用flood_fill染色一下就好了。另外需要吐槽一下的是，我同样的代码用C++写就是全部AC，最慢的点是88ms，但是拿Java写就TLE了三个。。。

代码

#include <cstring>
#include <iostream>
#include <cmath>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <iomanip>
#include <vector>

using namespace std;

const int maxn = 1010;
int n, m;
char mazes[maxn][maxn];
int colors[maxn][maxn];
int cell_nums[maxn * maxn];
int color = 0;
int dx[] = {-1, 1, 0, 0};
int dy[] = {0, 0, -1, 1};

void flood_fill(int x, int y) {
if(colors[x][y] != -233) {
return;
}
colors[x][y] = color;
for(int i = 0; i < 4; i++) {
int temp_x = x + dx[i];
int temp_y = y + dy[i];
if(temp_x < 0 || temp_x >= n) {
continue;
}
if(temp_y < 0 || temp_y >= n) {
continue;
}
if(mazes[x][y] != mazes[temp_x][temp_y]) {
flood_fill(temp_x, temp_y);
}
}
}

int main() {
scanf("%d %d", &n, &m);
for(int i = 0; i < n; i++) {
scanf("%s", mazes[i]);
}
for(int i = 0; i < n; i++) {
for(int j = 0; j < n; j++) {
colors[i][j] = -233;
}
}
for(int i = 0; i < n; i++) {
for(int j = 0; j < n; j++) {
if(colors[i][j] == -233) {
color++;
}
flood_fill(i, j);
}
}
for(int i = 0; i < maxn * maxn; i++) {
cell_nums[i] = 0;
}
for(int i = 0; i < n; i++) {
for(int j = 0; j < n; j++) {
cell_nums[colors[i][j]]++;
}
}
for(int i = 0; i < m; i++) {
int x, y;
scanf("%d %d", &x, &y);
printf("%d\n", cell_nums[colors[x-1][y-1]]);
}
return 0;
}