[LeetCode] BFS解决的题目

一.130  Surrounded Regions(https://leetcode.com/problems/surrounded-regions/description/)

题目:

《[LeetCode] BFS解决的题目》

解法:

这道题的意思是将所有被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/)

题目:

《[LeetCode] BFS解决的题目》

 

解法:

这道题的基本思想跟上面一道差不多,也是用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);             
        }
    }
};

 

 三.542 01 Matrix

 题目:

《[LeetCode] BFS解决的题目》

 

解法:

《[LeetCode] BFS解决的题目》

代码:

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        vector<vector<int> > ans = matrix;
        queue<int> x, y;
        for (int i = 0; i < ans.size(); i++) {
            for (int j = 0; j < ans[0].size(); j++) {
                if (matrix[i][j] == 0) {
                    ans[i][j] = 0;
                    x.push(i);
                    y.push(j);
                }
                else {
                    ans[i][j] = -1;
                } 
            }
        }
        while (!x.empty() && !y.empty()) {
            int i = x.front(), j = y.front();
            x.pop();
            y.pop();
            if (i - 1 >= 0 && ans[i - 1][j] == -1) {
                ans[i - 1][j] = ans[i][j] + 1;
                x.push(i - 1);
                y.push(j);
            }
            if (i + 1 < ans.size() && ans[i + 1][j] == -1) {
                  ans[i + 1][j] = ans[i][j] + 1;
                  x.push(i + 1);
                y.push(j);
            }
            if (j - 1 >= 0 && ans[i][j - 1] == -1) {
                ans[i][j - 1] = ans[i][j] + 1;
                x.push(i);
                y.push(j - 1);
            }
            if (j + 1 < ans[0].size() && ans[i][j + 1] == -1) {
                ans[i][j + 1] = ans[i][j] + 1;
                x.push(i);
                y.push(j + 1);
            }
        }
        return ans;
    }
};

2018.1.7补充:
上面这一题,重做了一下,想到一个问题: 为什么可以不判断一下就直接ans[i][j] + 1呢?

题目要求离最近的0的距离,而BFS是一层层叠加的、从小到大的,在if处加了ans[i + 1][j] == -1的判断,说明要已修改的值是不能再修改的,满足离最近的0的距离的要求。(描述得不是很清楚,不过自己能回想起就对了)

 

LeetCode上面还有其他一些用BFS解决的问题,比如Word Ladder,但在前面已经写过了。

    原文作者:fengzw
    原文地址: https://www.cnblogs.com/fengziwei/p/7834529.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞