LeetCode 130 被围绕的区域 Java

给定一个二维的矩阵，包含 ‘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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路：无论是bfs还是dfs都应该从边缘开始遍历，因为只有与边缘连通的O，才不能被转化成X
1、从地图的四周开始bfs或者dfs
2、将与四周连通的O路径赋值成‘#’
3、遍历地图，将地图中的O赋值成X，将’#’赋值成O

dfs：

class Solution {
    int m,n;
	int vis[][];
	int dir[][] = {{0,1},{1,0},{0,-1},{-1,0}};
    public void solve(char[][] board) {
        m = board.length;
        if(m == 0) return;
		n = board[0].length;
		vis = new int[m][n];
		for(int i = 0; i < m; i++) {
			if(board[i][0] == 'O' && vis[i][0] == 0) {
				vis[i][0] = 1;
				board[i][0] = '#';
				dfs(i,0,board);
			}
		}
		for(int i = 0; i < m; i++) {
			if(board[i][n-1] == 'O' && vis[i][n-1] == 0) {
				vis[i][n-1] = 1;
				board[i][n-1] = '#';
				dfs(i,n-1,board);
			}
		}
		for(int j = 0; j < n; j++) {
			if(board[0][j] == 'O' && vis[0][j] == 0) {
				vis[0][j] = 1;
				board[0][j] = '#';
				dfs(0,j,board);
			}
		}
		for(int j = 0; j < n; j++) {
			if(board[m-1][j] == 'O' && vis[m-1][j] == 0) {
				vis[m-1][j] = 1;
				board[m-1][j] = '#';
				dfs(m-1,j,board);
			}
		}
		
		for(int i = 0; i < m; i++) {
			for(int j = 0; j < n; j++) {
				if(board[i][j] == 'O') {
					board[i][j] = 'X';
				}
				if(board[i][j] == '#') {
					board[i][j] = 'O';
				}
			}
		}
	}
	private void dfs(int x, int y, char[][] board) {
		for(int i = 0; i < 4; i++) {
			int nx = x + dir[i][0];
			int ny = y + dir[i][1];
			if(nx >= 0 && nx < m && ny >= 0 && ny < n && vis[nx][ny] == 0 && board[nx][ny] == 'O') {
				vis[nx][ny] = 1;
				board[nx][ny] = '#';
				dfs(nx,ny,board);
			}
		}
    }
}

bfs：

import java.util.LinkedList;
import java.util.Queue;
class Solution {
    public void solve(char[][] board) {
        Queue<Integer> queue = new LinkedList<Integer>();
		int n = board.length;
        if(n == 0) {
            return;
        }
		int m = board[0].length;
        int vis[][] = new int[n][m];
		for(int i = 0; i < n; i++) {
			int t = 0;
			
			if(board[i][m-1] == 'O') {
				t = i*m+m-1;
				queue.add(t);
                vis[i][m-1] = 1;
			}
			if(board[i][0] == 'O') {
				t = i*m+0;
				queue.add(t);
                vis[i][0] = 1;
			}
			
		}
		for(int i = 0; i < m; i++) {
			int t = 0;
			if(board[0][i] == 'O') {
				t = 0*m+i;
				queue.add(t);
                vis[0][i] = 1;
			}
			if(board[n-1][i] == 'O') {
				t = (n-1)*m+i;
				queue.add(t);
                vis[n-1][i] = 1;
			}
		}
		if(queue.isEmpty()) {
			for(int i = 0; i < n; i++) {
				for(int j = 0; j < m; j++) {
					board[i][j] = 'X';
				}
			}
			
		}
		int[][] dir = {{0,1},{1,0},{-1,0},{0,-1}};
		
		while(!queue.isEmpty()) {
			int val = queue.poll();
			int x = val/m;
			int y = val%m;
			
			for(int i = 0; i < 4; i++) {
				int nx = x + dir[i][0];
				int ny = y + dir[i][1];
				if(nx >= 0 && nx < n && ny >= 0 && ny < m && vis[nx][ny] == 0) {
					if(board[nx][ny] == 'O') {
						vis[nx][ny] = 1;
						queue.add(nx*m+ny);
					}
				}
			}
		}
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < m; j++) {
				if(board[i][j] == 'O' && vis[i][j] == 0) {
					board[i][j] = 'X';
				}
			}
		}
    }
}