【LeetCode】被围绕的区域

题目描述

给定一个二维的矩阵，包含 ‘X’ 和 ‘O’（字母 O）。
找到所有被 ‘X’ 围绕的区域，并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。
示例:

X X X X
X O O X
X X O X
X O X X

运行你的函数后，矩阵变为：

X X X X
X X X X
X X X X
X O X X

解释:
被围绕的区间不会存在于边界上，换句话说，任何边界上的 ‘O’ 都不会被填充为 ‘X’。 任何不在边界上，或不与边界上的 ‘O’ 相连的 ‘O’ 最终都会被填充为 ‘X’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

解题思路

先找到边界上的O，再依次寻找与边界相连的O，将这些O先用W标记，直至所有的边界判断完之后，遍历原矩阵，若是O的位置就改为X，若是W的位置就改为O即可。

完整代码

int pos[4][2] = { { 0, 1 }, { 1, 0 }, { 0, -1 }, { -1, 0 } };  //上下左右
class Solution {
public:
	void dfs(vector<vector<char>>& board, int x, int y, int row, int col)
	{
		board[x][y] = 'W';
		for (int k = 0; k < 4; k++)
		{
			int nx = x + pos[k][0];
			int ny = y + pos[k][1];
			if (nx<0 || nx >= row || ny<0 || ny >= col)
				continue;
			if (board[nx][ny] == 'O')
				dfs(board, nx, ny, row, col);
		}
	}

	void solve(vector<vector<char>>& board) {
		if (board.empty())
			return;
		int row = board.size();
		int col = board[0].size();

		for (int j = 0; j<col; ++j)
		{
			if (board[0][j] == 'O')
				dfs(board, 0, j, row, col);
			if (board[row - 1][j] == 'O')
				dfs(board, row - 1, j, row, col);
		}
		for (int i = 0; i<row; ++i)
		{
			if (board[i][0] == 'O')
				dfs(board, i, 0, row, col);
			if (board[i][col - 1] == 'O')
				dfs(board, i, col - 1, row, col);
		}
		for (int i = 0; i<row; i++)
		{
			for (int j = 0; j<col; j++)
			{
				if (board[i][j] == 'O')
					board[i][j] = 'X';
				else if (board[i][j] == 'W')
					board[i][j] = 'O';
			}
		}
	}

};

原题链接

链接：https://leetcode-cn.com/problems/surrounded-regions