OJ刷题记录：L1-508-围棋(20分)

L1-508-围棋(20分)

题目要求：
给定一个二维的矩阵，包含 ‘X’ 和 ‘O’（字母 O）。 找到所有被 ‘X’ 围绕的区域，并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。被围绕的区间不会存在于边界上，换句话说，任何边界上的 ‘O’ 都不会被填充为 ‘X’。 任何不在边界上，或不与边界上的 ‘O’ 相连的 ‘O’ 最终都会被填充为 ‘X’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。
输入
第一行输入矩阵的大小n，m，其中n,m < 100
接下来第二行输入n行m列的二维矩阵
输出
输出最终的二维矩阵
样例输入
4 4
X X X X
X O O X
X X O X
X O X X
样例输出
XXXX
XXXX
XXXX
XOXX

解题思路：
广度优先搜索（BFS）。在对数组进行遍历搜索时替换 O 为 X，若搜索时没有遇到处于边界的 O，则说明 O 被 X 围住，替换完成；若在搜索时遇到处于边界的 O，说明替换是无效的，则将数组复原，退出遍历。

通关代码：

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

void cheakIsland(vector<vector<char>> &matrix, int originX, int originY) {
	const int bx[] = {0, 0, -1, 1};
	const int by[] = {1, -1, 0, 0};
	vector<vector<char>> previous = matrix;
	
	matrix[originX][originY] = 0;
	
	queue<pair<int, int>> q;
	q.emplace(pair<int, int>(originX, originY));
	matrix[originX][originY] = 'X';
	
	int x, y;
	for (; !q.empty();) {
		x = q.front().first;
		y = q.front().second;
		q.pop();
		
		for (int i = 0; i < 4; i++) {
			if (x + bx[i] >= 0 && x + bx[i] < matrix.size() && y + by[i] >= 0 && y + by[i] < matrix[0].size()) {
				if (matrix[x + bx[i]][y + by[i]] == 'O') {
					matrix[x + bx[i]][y + by[i]] = 'X';
					q.emplace(pair<int, int>(x + bx[i], y + by[i]));
					if (x + bx[i] == 0 || x + bx[i] == matrix.size() - 1 || y + by[i] == 0 || y + by[i] == matrix[0].size() - 1) {
						matrix = previous;
						return;
					}
				}
			}
		}
	}	
}

int main() {
	int row, column;
	
	cin >> row >> column;

	vector<char> arr(column);
	vector<vector<char>> matrix;
	
	for (int i = 0; i < row; i++) {
		for (int j = 0; j < column; j++) {
			cin >> arr[j];
		}
		matrix.emplace_back(arr);
	}
	
	for (int i = 0; i < row; i++) {
		for (int j = 0; j < column; j++) {
			if (matrix[i][j] == 'O') {
				if (i == 0 || i == matrix.size() - 1 || j == 0 || j == matrix[0].size() - 1) {
					continue;
				}
				cheakIsland(matrix, i, j);
			}
		}
	}
	
	for (int i = 0; i < row; i++) {
		for (int j = 0; j < column; j++) {
			cout << matrix[i][j];
		}
		if (i != row - 1) {
			cout << endl;
		}
	}
	
	return 0;
} 

通关截图：