LeetCode【每日一题】_130. 被围绕的区域_java

问题描述

130.被围绕的区域

给定一个二维的矩阵，包含 ‘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’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

题解

这道题是要求把所有被X包围的O变成X，但是直接判断某个O是否被包围不太容易，所有可以先从边界入手。

1. 先将边界上的所有O以及与它们相连接的所有O标记一下，剩下的O一定是被包围的。
2. 然后遍历整个数组，
   当遇到O时，将它改为X；
   当遇到标记时，将它还原成O；

深度优先搜索

用深度优先搜索进行标记

class Solution {
    int n, m;
	public void solve(char[][] board) {
		n = board.length;
        if (n == 0) return;
		m = board[0].length;
		for (int i = 0; i < n; i++) {
			dfs(board, i, 0);
			dfs(board, i, m - 1);
		}
		for (int i = 1; i < m - 1; i++) {
			dfs(board, 0, i);
			dfs(board, n - 1, i);
		}
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				if (board[i][j] == 'O') {
					board[i][j] = 'X';
				} else if (board[i][j] == 'A') {
					board[i][j] = 'O';
				}
			}
		}
    }

	private void dfs(char[][] board, int i, int j) {
		if (i < 0 || i >= n || j < 0 || j >= m) {
			return;
		}
		if (board[i][j] == 'O') {
			board[i][j] = 'A';
			dfs(board, i + 1, j);
			dfs(board, i - 1, j);
			dfs(board, i, j + 1);
			dfs(board, i, j - 1);
		}
    }
}

广度优先搜索

用广度优先搜索进行标记

class Solution {
	public void solve(char[][] board) {
		int n = board.length;
		if (n == 0) return;
		int m = board[0].length;
		Queue<int[]> queue = new LinkedList<int[]>();
		int[] dx = {1, -1, 0, 0},
			  dy = {0, 0, 1, -1};
		for (int i = 0; i < n; i++) {
			if (board[i][0] == 'O') {
				queue.offer(new int[] {i, 0});
			}
			if (board[i][m - 1] == 'O') {
				queue.offer(new int[] {i, m - 1});
			}
		}
		for (int i = 0; i < m; i++) {
			if (board[0][i] == 'O') {
				queue.offer(new int[] {0, i});
			}
			if (board[n - 1][i] == 'O') {
				queue.offer(new int[] {n - 1, i});
			}
		}
		while (!queue.isEmpty()) {
			int[] poll = queue.poll();
			int x = poll[0],
				y = poll[1];
			board[x][y] = 'A';
			for (int i = 0; i < 4; i++) {
				int tempX = x + dx[i],
					tempY = y + dy[i];
				if (tempX < 0 || tempX >= n || tempY < 0 || tempY >= m) {
					continue;
				}
				if (board[x + dx[i]][y + dy[i]] == 'O') {
					queue.offer(new int[] {x + dx[i], y + dy[i]});
				}
			}
		}
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				if (board[i][j] == 'O') {
					board[i][j] = 'X';
				} else if (board[i][j] == 'A') {
					board[i][j] = 'O';
				}
			}
		}
    }
}