题目:
给定一个二维的矩阵,包含 '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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
这是图问题。对于图问题,DFS/BFS必备法宝233333。
这个题目自己实现的时候使用的是DFS,当然使用BFS也可。
在解决的时候,如果总是想着:找出一个点,这个点是'O'而且不在边界上。
那么必定需要对整个board进行DFS,那样的时间复杂度很高,最高可达到O(n^4)的样子。
所以需要反向思维:
①从四条边界向内进行DFS,并且在DFS的同时打标签。
②处理完四条边界后,对未被打上标签的'O'位置进行转换,将其变为‘X’。
这样即可。
因为需要打标签,所以整体空间复杂度为O(n)。时间复杂度最大为O(n^2)的样子。
实现:
java版:(其中solve1是正确的解法,solve()是复杂度很高,而且不正确的解法,可以忽略23333)
package leetcode;
import java.util.Arrays;
/*
USER:LQY
DATE:2020/8/11
TIME:10:52
*/
public class leetcode_130 {
public static void main(String []args){
char [][]board = {
// {'X', 'X', 'X', 'X'},
// {'X', 'O', 'O', 'X'},
// {'X', 'X', 'O', 'X'},
// {'X', 'O', 'X', 'X'}
{'O','O','O'},
{'O','O','O'},
{'O','O','O'}
// {'O','O','0','O','X','X'},
// {'O','O','O','O','O','O'},
// {'O','X','O','X','O','O'},
// {'O','X','O','O','X','O'},
// {'O','X','O','X','O','O'},
// {'O','X','O','O','O','O'}
};
new leetcode_130().solve1(board);
for(int i = 0;i < board.length;i++){
for(int j = 0;j < board[0].length;j++)
System.out.print(board[i][j]+" ");
System.out.println();
}
}
int r,c;
public void solve1(char [][]board){
r = board.length;
if(r == 0) return;
c = board[0].length;
int [][]book = new int[r][c];
int flag = 1;
for(int i = 0;i < c;i++){
if(board[0][i] == 'X') continue;
if(book[0][i] != 0) continue; //已经被标记过
book[0][i] = (flag++);
merge(board, book, book[0][i], 0, i);
}
for(int i = 1;i < r;i++){
if(board[i][0] == 'X') continue;
if(book[i][0] != 0) continue; //已经被标记过
book[i][0] = (flag++);
merge(board, book, book[i][0], i, 0);
}
for(int i = 1;i < c;i++){
if(board[r-1][i] == 'X') continue;
if(book[r-1][i] != 0) continue; //已经被标记过
book[r-1][i] = (flag++);
merge(board, book, book[r-1][i], r-1, i);
}
for(int i = 1;i < r-1;i++){
if(board[i][c-1] == 'X') continue;
if(book[i][c-1] != 0) continue; //已经被标记过
book[i][c-1] = (flag++);
merge(board, book, book[i][c-1], i, c-1);
}
for(int i = 0;i < r;i++){
for(int j = 0;j < c;j++){
if(book[i][j]==0 && board[i][j]=='O') board[i][j] = 'X';
}
// System.out.println();
}
}
void merge(char [][]board, int [][]book, int flag, int nr, int nc){
if(nr>=r || nr<0 || nc>=c || nc<0) return;
if(board[nr][nc] == 'X') return;
book[nr][nc] = flag;
for(int i = 0;i < 4;i++){
int rr = nr + dir[i][0];
int rc = nc + dir[i][1];
if(rr>=r || rr<0 || rc>=c || rc<0) continue;
if(book[rr][rc] != 0) continue;
if(board[rr][rc] == 'X') continue;
merge(board, book, flag, rr, rc);
}
}
public void solve(char[][] board) {
r = board.length;
if(r == 0) return;
c = board[0].length;
int [][]book = new int[r][c];
for(int i = 1;i < r;i++){
for(int k = 0;k < r;k++)
Arrays.fill(book[k], 0);
for(int j = 1;j < c;j++){
if(board[i][j] == 'X') continue;
book[i][j] = 1;
if(find(board, i, j, book)){
System.out.println("find at ("+i+","+j+")");
board[i][j] = 'X';
}
}
}
}
int [][]dir = {{-1,0},{0,1},{1,0},{0,-1}};
public boolean find(char [][]board, int nr, int nc, int [][]book){
if(nr>=r || nr<0 || nc>=c || nc<0) return false;
// if(board[nr][nc]=='X' && (nr>0 && nr<r-1 && nc>0 && nc<c-1)) return true;
if((nr==r-1 || nr==0 || nc==c-1 || nc==0) && board[nr][nc]=='O') return false;
boolean ans = true;
for(int i = 0;i < 4;i++){
int rr = nr + dir[i][0];
int rc = nc + dir[i][1];
if(rr>=r || rr<0 || rc>=c || rc<0) continue;
if(board[rr][rc] == 'X') continue;
if(book[rr][rc] == 1) continue;
book[rr][rc] = 1;
ans = ans && find(board, rr, rc, book);
book[rr][rc] = 0;
}
return ans;
}
}