码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode] BFS解决的题目

时间:2017-11-14 21:20:12      阅读:179      评论:0      收藏:0      [点我收藏+]

标签:images   problems   优先   就是   思想   ges   amp   ems   while   

一.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,但在前面已经写过了。

[LeetCode] BFS解决的题目

标签:images   problems   优先   就是   思想   ges   amp   ems   while   

原文地址:http://www.cnblogs.com/fengziwei/p/7834529.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!