题目地址:https://leetcode-cn.com/probl… 题目描述:给定一个二维的矩阵,包含 ‘X’ 和 ‘O’(字母 O)。
找到所有被 ‘X’ 围绕的区域,并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。
示例:
X X X XX O O XX X O XX O X X 运行你的函数后,矩阵变为:
X X X XX X X XX X X XX 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;
}
}