由于没有在界面上找到“回复”这个功能选项,所以我就干脆重新在问题这儿编辑了,也不知道bobo老师能不能看到~
前几天用bfs得到了一个accepted之后,还是想着能不能改进一下dfs的代码,然后发现,我之前的那段代码的逻辑是存在问题的!!!
比如一个很简单的测试用例:
XXXX XXXX XXXX XOOX 用原问题中的代码会得到 XOXX 正确的应该是 XOOX XOXX ----------------------> XOXX ------------> XOXX XOXX XOXX XOXX
也就是说有一些"O"点会被误认为是被surrounded的点,具体的和d[4][2]中的值的顺序有关,但不管顺序如何,都会有这种情况发生。
所以,还是得从边界出发来往内部dfs,虽然这样遍历的情况会变多,但也没办法,下面是我改进后的代码,虽然仍旧无法解决栈溢出的问题,但至少这次的逻辑应该是没问题了~
class Solution { private: int m; int n; vector<vector<bool>> visited; int d[4][2] = {{0,1}, {1,0}, {0,-1}, {-1,0}}; private: bool inArea(int x, int y){ return x >= 0 && y >= 0 && x < m && y < n; } void dfs(vector<vector<char>> &board, int x, int y){ visited[x][y] = true; for (int i = 0; i < 4; i++){ int newX = x + d[i][0]; int newY = y + d[i][1]; if (!inArea(newX, newY)) continue; if (board[newX][newY] == 'O' && !visited[newX][newY]) dfs(board, newX, newY); } return; } public: void solve(vector<vector<char>>& board) { m = board.size(); if (m <= 2) return; n = board[0].size(); if (n <= 2) return; visited = vector<vector<bool>>(m, vector<bool>(n, false)); for (int i = 1; i < m - 1; i++){ if (board[i][0] == 'O' && !visited[i][0]) dfs(board, i, 0); if (board[i][n - 1] == 'O' && !visited[i][n - 1]) dfs(board, i, n - 1); } for (int i = 1; i < n - 1; i++){ if (board[0][i] == 'O' && !visited[0][i]) dfs(board, 0, i); if (board[m - 1][i] == 'O' && !visited[m - 1][i]) dfs(board, m - 1, i); } for (int i = 1; i < m - 1; i++){ for (int j = 1; j < n - 1; j++){ if (board[i][j] == 'O' && !visited[i][j]) board[i][j] = 'X'; } } return; } };
-------------------------------------------原问题如下-------------------------------------------
如题,Submit Solution会报Runtime Error,但我把那个例子复制下来单独Run Code没问题~
不知道是怎么回事,代码如下:
class Solution { private: int m; int n; vector<vector<bool>> visited; int d[4][2] = {{0,1}, {1,0}, {0,-1}, {-1,0}}; private: bool inArea(int x, int y){ return x >= 0 && y >= 0 && x < m && y < n; } bool dfs(vector<vector<char>> &board, int x, int y){ visited[x][y] = true; board[x][y] = 'X'; for (int i = 0; i < 4; i++){ int newX = x + d[i][0]; int newY = y + d[i][1]; if (!inArea(newX, newY)){ board[x][y] = 'O'; return false; } if (board[newX][newY] == 'O' && !dfs(board, newX, newY)){ board[x][y] = 'O'; return false; } } return true; } public: void solve(vector<vector<char>>& board) { m = board.size(); if (m <= 0) return; n = board[0].size(); visited = vector<vector<bool>>(m, vector<bool>(n, false)); int changedRegion = 0; for (int i = 0; i < m; i++){ for (int j = 0; j < n; j++){ if (board[i][j] == 'O' && !visited[i][j]) if(dfs(board, i, j)) changedRegion++; } } return; } };