Given a 2D board containing 'X'
and 'O'
, capture all regions surrounded by 'X'
.
A region is captured by flipping all 'O'
s into 'X'
s in that surrounded region.
For example,
X X X X X O O X X X O X X O X X
After running your function, the board should be:
X X X X X X X X X X X X X O X X
This is a simple problem, we just need to apply bfs on 'O' elements on the edges of board.
public class Solution { public void solve(char[][] board) { int row = board.length; if(row > 0){ int column = board[0].length; Set<Integer> alreadyChecked = new HashSet<Integer>(); for(int i = 0; i < column; ++i){ if(board[0][i] == 'O' && !alreadyChecked.contains(i)) bfs(board,0, i, alreadyChecked); if(board[row - 1][i] == 'O' && !alreadyChecked.contains((row - 1) * column + i)) bfs(board,row - 1, i, alreadyChecked); } for(int i = 1; i < row - 1; ++i){ if(board[i][0] == 'O' && !alreadyChecked.contains(i * column)) bfs(board,i, 0, alreadyChecked); if(board[i][column - 1] == 'O' && !alreadyChecked.contains(i * column + column - 1)) bfs(board,i, column - 1, alreadyChecked); } for(int i = 0; i < row; ++i){ for(int j = 0; j < column; ++j){ if(board[i][j] == 'O' && !alreadyChecked.contains(i*column + j)) board[i][j] = 'X'; } } } } private void bfs(char[][] board, int i, int j, Set<Integer> alreadyChecked) { int row = board.length; if(row > 0){ int column = board[0].length; if(!alreadyChecked.contains(i * column + j)) { Queue<Integer> current = new LinkedList<Integer>(); alreadyChecked.add(i * column + j); current.add(i * column + j); while(current.peek() != null){ int currentCoordinate = current.poll(); int x = currentCoordinate / column; int y = currentCoordinate % column; if(x - 1 >= 0 && board[x - 1][y] == 'O' && !alreadyChecked.contains(currentCoordinate - column)){ current.add(currentCoordinate - column); alreadyChecked.add(currentCoordinate - column); } if(x + 1 < row && board[x + 1][y] == 'O' && !alreadyChecked.contains(currentCoordinate + column)){ current.add(currentCoordinate + column); alreadyChecked.add(currentCoordinate + column); } if(y - 1 >= 0 && board[x][y - 1] == 'O' && !alreadyChecked.contains(currentCoordinate - 1)) { current.add(currentCoordinate - 1); alreadyChecked.add(currentCoordinate - 1); } if(y + 1 < column && board[x][y + 1] == 'O' && !alreadyChecked.contains(currentCoordinate + 1)) { current.add(currentCoordinate + 1); alreadyChecked.add(currentCoordinate + 1); } } } } } }