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

【leetcode】Surrounded Regions(middle)☆

时间:2015-05-03 21:53:42      阅读:205      评论:0      收藏:0      [点我收藏+]

标签:

Given a 2D board containing ‘X‘ and ‘O‘, capture all regions surrounded by ‘X‘.

A region is captured by flipping all ‘O‘s into ‘X‘s in that surrounded region.

For example,

X X X X
X O O X
X X O X
X O X X

 

After running your function, the board should be:

X X X X
X X X X
X X X X
X O X X

 

思路:

反向思考,先找到没有被包围的区域,标记为‘+’,再把标为‘+’的区域标为‘O‘,标为’O‘的区域改为‘X‘。没有被包围的区域一定是与最外圈的’O‘相连的区域,所以要先遍历区域的上下左右边界,找到‘O‘的地方。接下来的问题是如何把所有与最外层‘O‘相连的区域标记上。有两种思路:BFS和DFS。

所谓BFS是指把当前标记位置的上下左右都标记一遍,然后再标记相邻点的上下左右位置。不需递归,用队列。

DFS是指把当前标记位置的向一个方向标记,比如一直向左,直到没有可标记的,再换一个方向。需要递归。

代码里面BFS可以通过,DFS栈溢出了。

void solve(vector<vector<char>> &board) {
        if(board.size() == 0) return;
        int rowNum = board.size();
        int colNum = board[0].size();
        //遍历最外面一圈,找‘O‘
        //最上
        for(int j = 0; j < colNum; j++)
        {
            if(board[0][j] == O)
                BFS(board, 0, j);
        }
        //最下
        for(int j = 0; j < colNum; j++)
        {
            if(board[rowNum - 1][j] == O)
                BFS(board, rowNum - 1, j);
        }
        //最左
        for(int i = 0; i < rowNum; i++)
        {
            if(board[i][0] == O)
                BFS(board, i, 0);
        }
        //最右
        for(int i = 0; i < rowNum; i++)
        {
            if(board[i][colNum - 1] == O)
                BFS(board, i, colNum - 1);
        }

        for(int i = 0; i < rowNum; i++)
        {
            for(int j = 0; j < colNum; j++)
            {
                if(board[i][j] == O)
                    board[i][j] = X;
                if(board[i][j] == +)
                    board[i][j] = O;
            }
        }
        
    }
    void DFS(vector<vector<char>> &board, int r, int c)
    {
        if(r >= 0 && c >= 0 && r < board.size() && c < board[0].size() && board[r][c] == O)
        {
            board[r][c] = +;
            DFS(board, r - 1, c);
            DFS(board, r + 1, c);
            DFS(board, r, c - 1);
            DFS(board, r, c + 1);
        }
    }
    void BFS(vector<vector<char>> &board, int r, int c)
    {
        queue<pair<int, int>> q;
        q.push(make_pair(r, c));
        while(!q.empty())
        {
            int i = q.front().first;
            int j = q.front().second;
            q.pop();
            if(i >= 0 && j >= 0 && i < board.size() && j < board[0].size() && board[i][j] == O)
            {
                board[i][j] = +;
                q.push(make_pair(i - 1, j));
                q.push(make_pair(i + 1, j));
                q.push(make_pair(i, j - 1));
                q.push(make_pair(i, j + 1));
            }
            
        }

    }

 

上面的代码已经是优化过的了,我自己写的时候只写出了DFS的,而且自己也没有意识到是DFS。代码也很繁琐。注意通过把判断条件放在一起来简化代码。

我原本很挫的代码:栈溢出。

class Solution {
public:
    void solve(vector<vector<char>> &board) {
        if(board.size() == 0) return;
        int rowNum = board.size();
        int colNum = board[0].size();
        //遍历最外面一圈,找‘O‘
        //最上
        for(int j = 0; j < colNum; j++)
        {
            if(board[0][j] == O)
            {
                board[0][j] = +;
                mySolve(board, 0, j);
            }
        }
        //最下
        for(int j = 0; j < colNum; j++)
        {
            if(board[rowNum - 1][j] == O)
            {
                board[rowNum - 1][j] = +;
                mySolve(board, rowNum - 1, j);
            }
        }
        //最左
        for(int i = 0; i < rowNum; i++)
        {
            if(board[i][0] == O)
            {
                board[i][0] = +;
                mySolve(board, i, 0);
            }
        }
        //最右
        for(int i = 0; i < rowNum; i++)
        {
            if(board[i][colNum - 1] == O)
            {
                board[i][colNum - 1] = +;
                mySolve(board, i, colNum - 1);
            }
        }

        for(int i = 0; i < rowNum; i++)
        {
            for(int j = 0; j < colNum; j++)
            {
                if(board[i][j] == O)
                    board[i][j] = X;
                if(board[i][j] == +)
                    board[i][j] = O;
            }
        }
        
    }
    void mySolve(vector<vector<char>> &board, int r, int c)
    {
        if(r - 1 >= 0 && board[r - 1][c] == O) //
        {
            board[r - 1][c] == +;
            mySolve(board, r - 1, c);
        }
        if(r + 1 < board.size() && board[r + 1][c] == O) //
        {
            board[r + 1][c] == +;
            mySolve(board, r + 1, c);
        }
        if(c - 1 >= 0 && board[r][c - 1] == O) //
        {
            board[r][c - 1] == +;
            mySolve(board, r, c - 1);
        }
        if(c + 1 < board[0].size() && board[r][c + 1] == O) //
        {
            board[r][c + 1] == +;
            mySolve(board, r, c + 1);
        }
    }
};

 

【leetcode】Surrounded Regions(middle)☆

标签:

原文地址:http://www.cnblogs.com/dplearning/p/4474768.html

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