一.130 Surrounded Regions(https://leetcode.com/problems/surrounded-regions/description/)
题目:
解法:
这道题的意思是将所有被X包围的O都变为X(边缘的不算),我们可以维护一个队列,先把四周的O的位置放进队列中,并把这个位置的值变为Y。然后每次从队列中拿出一个位置,把这个位置四周为O的位置的值变为Y,再把这个位置放进队列(为什么要先变值再进队列呢?在下一道题中会说)。一直到队列为空时,我们就成功地把所有直接或间接地与边缘的O相连的其他O变为了Y,剩余的O就是被X包围的了。然后遍历一遍,把为O的位置变为X,为Y的位置变为O,就得出结果了。
这道题利用到了BFS(宽度优先搜索)的思想,那么用深度优先是否可以呢?据说会TLE,因为图可能很大,递归会导致栈溢出。但是,我认为分别用这两种不同的方法思考一下这道题,可以锻炼利用bfs和dfs解题的思想。
下面是代码:
class Solution { public: void solve(vector<vector<char>>& board) { queue<int> x; queue<int> y; 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‘) { x.push(i); y.push(j); board[i][j] = ‘Y‘; } } } while (!x.empty() && !y.empty()) { int i = x.front(); int j = y.front(); x.pop(); y.pop(); if (i - 1 >= 0 && board[i - 1][j] == ‘O‘) { x.push(i - 1); y.push(j); board[i - 1][j] = ‘Y‘; } if (i + 1 < board.size() && board[i + 1][j] == ‘O‘) { x.push(i + 1); y.push(j); board[i + 1][j] = ‘Y‘; } if (j - 1 >= 0 && board[i][j - 1] == ‘O‘) { x.push(i); y.push(j - 1); board[i][j - 1] = ‘Y‘; } if (j + 1 < board[0].size() && board[i][j + 1] == ‘O‘) { x.push(i); y.push(j + 1); board[i][j + 1] = ‘Y‘; } } for (int i = 0; i < board.size(); i++) { for (int j = 0; j < board[i].size(); j++) { if (board[i][j] == ‘Y‘) { board[i][j] = ‘O‘; } else { board[i][j] = ‘X‘; } } } } };
二. 200 Number of Islands(https://leetcode.com/problems/number-of-islands/description/)
题目:
解法:
这道题的基本思想跟上面一道差不多,也是用bfs,维护一个队列:
当遇到1的时候,把它改为0,进队列;(每次遇到新的island时)
每次把一个位置出队列,并把这个位置四周为1的位置改为0,然后把这些位置放入队列中。当队列为空时,代表已经找到一个island的所有1了,结果+1。
解法是简单的,但我做题的时候还是踩了不少坑:
(1)我用了三层循环去写,因为我想着要判断什么时候所有位置都为0,但其实是多此一举,只需要遍历每一个位置就行了。
(2)我一开始是先把位置放进队列,在出队列的时候才把这个位置的值变为0,但这是不对的:这样会造成一个位置的多次进队列(细想一下吧,出队列的时候才变为0,那在进队列到出队列的过程中这个位置的值还是1,还是会被其他位置检测到,多次加进队列),肯定会TLE。因此一定要先把值变为0,再加进队列。
代码如下:
class Solution { public: int numIslands(vector<vector<char> >& grid) { int ans = 0; queue<int> x, y; for (int i = 0; i < grid.size(); i++ ) { for (int j = 0; j < grid[0].size(); j++) { if (grid[i][j] == ‘1‘) { x.push(i); y.push(j); grid[i][j] = ‘0‘; while (!x.empty() && !y.empty()) { int i = x.front(); int j = y.front(); x.pop(); y.pop(); if (i - 1 >= 0 && grid[i - 1][j] == ‘1‘) { x.push(i - 1); y.push(j); grid[i - 1][j] = ‘0‘; } if (i + 1 < grid.size() && grid[i + 1][j] == ‘1‘) { x.push(i + 1); y.push(j); grid[i + 1][j] = ‘0‘; } if (j - 1 >= 0 && grid[i][j - 1] == ‘1‘) { x.push(i); y.push(j - 1); grid[i][j - 1] = ‘0‘; } if (j + 1 < grid[0].size() && grid[i][j + 1] == ‘1‘) { x.push(i); y.push(j + 1); grid[i][j + 1] = ‘0‘; } } ans++; } } } return ans; } };
或者:
class Solution { public: int numIslands(vector<vector<char> >& grid) { int ans = 0; for (int i = 0; i < grid.size(); i++ ) { for (int j = 0; j < grid[0].size(); j++) { if (grid[i][j] == ‘1‘) { bfs(grid, i, j); ans++; } } } return ans; } void bfs(vector<vector<char> >& grid, int i, int j) { grid[i][j] = ‘0‘; if (i - 1 >= 0 && grid[i - 1][j] == ‘1‘) { bfs(grid, i - 1, j); } if (i + 1 < grid.size() && grid[i + 1][j] == ‘1‘) { bfs(grid, i + 1, j); } if (j - 1 >= 0 && grid[i][j - 1] == ‘1‘) { bfs(grid, i, j - 1); } if (j + 1 < grid[0].size() && grid[i][j + 1] == ‘1‘) { bfs(grid, i, j + 1); } } };
LeetCode上面还有其他一些用BFS解决的问题,比如Word Ladder,但在前面已经写过了。