【LeetCode】被围绕的区域

深碍√TFBOYSˉ_ 2021-11-01 11:20 349阅读 0赞

#### 题目描述 ####

给定一个二维的矩阵，包含 ‘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