题目链接
https://leetcode-cn.com/problems/surrounded-regions/
描述
给定一个二维的矩阵,包含 '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'。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
初始代码模板
class Solution {
public void solve(char[][] board) {
}
}
代码
初看晦涩难懂,其实转换个思路,就很简单了。
和边界相连的’O’肯定是无法填充的,所以可以把与边界相连的’O’区间先用特殊字符标识一下,比如换成’#’,然后遍历一下数组,不是’#‘的换成’X’,而’#‘改回’O’,就是答案了,没有必要寻找被围绕的区间。
class Solution {
private int[] xArr = {1, -1, 0, 0};
private int[] yArr = {0, 0, 1, -1};
public void solve(char[][] board) {
if (board == null || board.length == 0 || board[0] == null || board[0].length == 0) {
return;
}
int m = board.length;
int n = board[0].length;
for (int i = 0; i < n; i++) {
dfs(board, 0, i);
dfs(board, m - 1, i);
}
for (int i = 0; i < m; i++) {
dfs(board, i, 0);
dfs(board, i, n - 1);
}
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
board[i][j] = board[i][j] == '#' ? 'O' : 'X';
}
}
}
private void dfs(char[][] board, int x, int y) {
if (x < 0 || x >= board.length || y < 0 || y >= board[0].length || board[x][y] != 'O') {
return;
}
board[x][y] = '#';
for (int i = 0; i < 4; i++) {
dfs(board, x + xArr[i], y + yArr[i]);
}
}
}