題目地址:
https://leetcode-cn.com/probl…
題目描述:
給定一個二維的矩陣,包含 `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
解答:
就是找所有被`X`圍繞的區域,然後填充是輕而易舉的事情。
可以這樣考慮,一個O如果在邊界上那它一定不可能被`X`圍繞,並且如果一個O和
邊界上的O聯通(可達),那麼它也不可能被`X`圍繞。因此我們只需要找到邊界上
所有的O,然後對它進行深度優先搜尋,使得所有和它聯通的O(包括它自己)都被標記
然後剩下的未被標記的O就是被完全圍繞的,然後填充即可!
這一題考察的是圖的深度優先搜尋!
java ac程式碼:
class Solution {
public void solve(char[][] board) {
//利用並查集,只要某個點不和邊界的O點聯通,則它就是被包圍的。
//利用dfs進行連通性構建
if(board == null || board.length == 0)return;
int[] set = new int[board.length*board[0].length];
for(int i = 0;i < set.length;i++)
set[i] = i;
boolean[][] flag = new boolean[board.length][board[0].length];
for(int i = 0;i < board.length;i++)
for(int j = 0;j < board[0].length;j++)
if(board[i][j] == `O`&&!flag[i][j])
{
dfs(board,flag,set,i,j,loc(board,i,j));
}
HashSet<Integer>hashset = new HashSet(500);
for(int i = 0;i < board.length;i++)
{
if(board[i][0]==`O`)
hashset.add(find(set,loc(board,i,0)));
if(board[i][board[0].length-1]==`O`)
hashset.add(find(set,loc(board,i,board[0].length-1)));
}
for(int j = 0;j < board[0].length;j++)
{
if(board[0][j]==`O`)
hashset.add(find(set,loc(board,0,j)));
if(board[board.length-1][j]==`O`)
hashset.add(find(set,loc(board,board.length-1,j)));
}
for(int i = 1;i < board.length-1;i++)
for(int j = 1;j < board[0].length-1;j++)
if(board[i][j] == `O`&&!hashset.contains(find(set,loc(board,i,j)) ))
board[i][j] = `X`;
}
void dfs(char[][] board,boolean[][]flag,int[]set,int x,int y,int father)
{
if(!(x>=0&&x<board.length&&y>=0&&y<board[0].length)||board[x][y]!=`O`||flag[x][y])
return;
flag[x][y] = true;
set[loc(board,x,y)] = father;
dfs(board,flag,set,x-1,y,father);
dfs(board,flag,set,x+1,y,father);
dfs(board,flag,set,x,y-1,father);
dfs(board,flag,set,x,y+1,father);
}
int find(int[] set,int i)
{
if(set[i] == i)return i;
return set[i] = find(set,set[i]);
}
//做一個位置對映
int loc(char[][]board,int i,int j)
{
return i*board[0].length+j;
}
}