力扣(LeetCode)130

Linus脫襪子發表於2019-01-19

題目地址:
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;
   }
    
}



相關文章