給定一個(gè)二維的矩陣砾肺,包含 'X' 和 'O'(字母 O)齐佳。
找到所有被 'X' 圍繞的區(qū)域,并將這些區(qū)域里所有的 'O' 用 'X' 填充债沮。
示例:
X X X X
X O O X
X X O X
X O X X
運(yùn)行你的函數(shù)后炼吴,矩陣變?yōu)椋?X X X X
X X X X
X X X X
X O X X
解釋:
被圍繞的區(qū)間不會(huì)存在于邊界上,換句話說(shuō)疫衩,任何邊界上的 'O' 都不會(huì)被填充為 'X'硅蹦。 任何不在邊界上,
或不與邊界上的 'O' 相連的 'O' 最終都會(huì)被填充為 'X'闷煤。如果兩個(gè)元素在水平或垂直方向相鄰童芹,則稱它們是“相連”的。
代碼
class Solution {
public:
void solve(vector<vector<char> >& board) {
for (int i = 0; i < board.size(); ++i) {
for (int j = 0; j < board[i].size(); ++j) {
if ((i == 0 || i == board.size() - 1 || j == 0 || j == board[i].size() - 1) && board[i][j] == 'O')
solveDFS(board, i, j);
}
}
for (int i = 0; i < board.size(); ++i) {
for (int j = 0; j < board[i].size(); ++j) {
if (board[i][j] == 'O') board[i][j] = 'X';
if (board[i][j] == '$') board[i][j] = 'O';
}
}
}
void solveDFS(vector<vector<char> > &board, int i, int j) {
if (board[i][j] == 'O') {
board[i][j] = '$';
if (i > 0 && board[i - 1][j] == 'O')
solveDFS(board, i - 1, j);
if (j < board[i].size() - 1 && board[i][j + 1] == 'O')
solveDFS(board, i, j + 1);
if (i < board.size() - 1 && board[i + 1][j] == 'O')
solveDFS(board, i + 1, j);
if (j > 1 && board[i][j - 1] == 'O')
solveDFS(board, i, j - 1);
}
}
};