给你一个 m x n 的矩阵 board ,由若干字符 ‘X’ 和 ‘O’ ,找到所有被 ‘X’ 围绕的区域,并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。
示例 1:
输入:board = [[“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’。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
思路:
1、对于每一个边界上的 O,我们以它为起点,标记所有与它直接或间接相连的字母 O;
2、最后我们遍历这个矩阵,如果该字母被标记过,则该字母为没有被字母 X 包围的字母 O,我们将其还原为字母 O;
如果该字母没有被标记过,则该字母为被字母 X 包围的字母 O,我们将其修改为字母 X。
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
void dfs(vector < vector<char>>& graph, int i, int j) {
if (i < 0 || i >= graph.size() || j < 0 || j >= graph[0].size() || graph[i][j] != 'O')
{
return;
}
graph[i][j] = '*';
dfs(graph, i - 1, j);
dfs(graph, i + 1, j);
dfs(graph, i, j + 1);
dfs(graph, i, j - 1);
}
void solve(vector<vector<char>>& board) {
int m = board.size();
int n = board[0].size();
if (m == 0 || n == 0)
{
return;
}
for (int i = 0; i < m; i++)//标记边上的‘O'
{
for (int j = 0; j < n; j++)
{
if ((i == 0 || j == 0 || i == m - 1 || j == n - 1) && board[i][j] == 'O')//越界或者不是'O'返回
{
dfs(board, i, j);
}
}
}
for (int i = 0; i < m; i++)//替换
{
for (int j = 0; j < n; j++)
{
if (board[i][j] == 'O')
{
board[i][j] = 'X';
}else if (board[i][j] == '*') {
board[i][j] = 'O';
}
}
}
}
int main() {
vector<vector<char>> boards{ {'X', 'X', 'X', 'X'},
{ 'X','O','O','X' },
{ 'X','X','O','X' },
{ 'X','O','X','X' } };
solve(boards);
for (int i = 0; i < boards.size(); i++)
{
for (int j = 0; j < boards[0].size(); j++) {
cout << boards[i][j] << " ";
}
cout << endl;
}
cout << endl;
return 0;
}