由于没有在界面上找到“回复”这个功能选项,所以我就干脆重新在问题这儿编辑了,也不知道bobo老师能不能看到~
前几天用bfs得到了一个accepted之后,还是想着能不能改进一下dfs的代码,然后发现,我之前的那段代码的逻辑是存在问题的!!!
比如一个很简单的测试用例:
1 2 3 4 | XXXX XXXX XXXX XOOX 用原问题中的代码会得到 XOXX 正确的应该是 XOOX XOXX ----------------------> XOXX ------------> XOXX XOXX XOXX XOXX |
也就是说有一些"O"点会被误认为是被surrounded的点,具体的和d[4][2]中的值的顺序有关,但不管顺序如何,都会有这种情况发生。
所以,还是得从边界出发来往内部dfs,虽然这样遍历的情况会变多,但也没办法,下面是我改进后的代码,虽然仍旧无法解决栈溢出的问题,但至少这次的逻辑应该是没问题了~
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | 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没问题~
不知道是怎么回事,代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | 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 ; } }; |