【dfs】被围绕的区域
2020-11-08 本文已影响0人
修行者12138
给定一个二维的矩阵,包含 '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'。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/surrounded-regions
AC代码
/**
* 上右下左
*/
int[][] direction = new int[][]{{-1, 0}, {0, 1}, {1, 0}, {0, -1}};
/**
* isNotSurrounded[i][j] = true表示board[i][j]没有被 'X' 围绕,默认为false
*/
boolean[][] isNotSurrounded;
char[][] board;
public void solve(char[][] board) {
if (board == null || board.length == 0 || board[0].length == 0) {
return ;
}
isNotSurrounded = new boolean[board.length][board[0].length];
this.board = board;
for (int j = 0; j < board[0].length; j++) {
// 第一行
if (board[0][j] == 'O' && (!isNotSurrounded[0][j])) {
isNotSurrounded[0][j] = true;
dfs(0, j);
}
// 最后一行
if (board[board.length - 1][j] == 'O' && (!isNotSurrounded[board.length - 1][j])) {
isNotSurrounded[board.length - 1][j] = true;
dfs(board.length - 1, j);
}
}
for (int i = 1; i < board.length - 1; i++) {
// 第一列
if (board[i][0] == 'O' && (!isNotSurrounded[i][0])) {
isNotSurrounded[i][0] = true;
dfs(i, 0);
}
// 最后一列
if (board[i][board[0].length - 1] == 'O' && (!isNotSurrounded[i][board[0].length - 1])) {
isNotSurrounded[i][board[0].length - 1] = true;
dfs(i, board[0].length - 1);
}
}
for (int i = 0; i < board.length; i++) {
for (int j = 0; j < board[0].length; j++) {
if (board[i][j] == 'O' && (!isNotSurrounded[i][j])) {
board[i][j] = 'X';
}
}
}
}
/**
* 把与board[i][j]相连的'O'标识为没有被 'X' 围绕
*/
private void dfs(int i, int j) {
for (int k = 0; k < direction.length; k++) {
int x = i + direction[k][0];
int y = j + direction[k][1];
if (check(x, y) && board[x][y] == 'O' && (!isNotSurrounded[x][y])) {
isNotSurrounded[x][y] = true;
dfs(x, y);
}
}
}
/**
* 判断board[i][j]是否合法
*/
private boolean check(int i, int j) {
return i >= 0 && i < board.length && j >= 0 && j < board[0].length;
}