[LeetCode] Matrix 值修改系列，例题 Surrounded Regions，Set Matrix Zeroes

引言

Matrix内部的值修改严格来讲放在一个系列里不大合适，因为对于不同的问题，所用的算法和技巧可能完全不同，权且这样归类，以后需要时再拆分吧。

 

例题 1

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 XX O O XX X O XX O X X

 

After running your function, the board should be:

X X X XX X X XX X X XX O X X

class Solution {public:    void solve(vector<vector<char>> &board) {    }};

这道题的思路应该比较容易想到：

遍历最外层一圈，如果有O，就把其相邻的也设置为O。接着遍历全矩阵，把内层O置为X。

但是这样做的问题遍历全矩阵时，分不清遇到的O是内层还是外层。

因此改进的方法是遍历最外层时，将O及其相邻的字符都设为"Y"。遍历全矩阵时，把"O"设置为X，把"Y"设置回"O"。

 

寻找O的邻居时，用的自然是BFS。每遇到一个O，就通过BFS将它以及邻居设置为"Y"

代码：

class Solution {struct Point{    int h;    int v;    Point(int vp, int hp) : v(vp), h(hp) {};};public:    void BFS(int startH, int startW, vector<vector<char>> &board, queue<Point> que){        while(!que.empty()) que.pop();        int W = board[0].size(), H = board.size();        Point p(startH, startW);        que.push(p);        while(!que.empty()){            Point cur = que.front();            que.pop();            board[cur.v][cur.h] = 'Y';            for(int i = 0; i < 4; i++){ //扫描四个方向上的邻居                if((cur.v+addV[i]) < H                && (cur.h+addH[i]) < W                && (cur.v+addV[i]) >= 0                && (cur.h+addH[i]) >= 0                 && board[cur.v+addV[i]][cur.h+addH[i]] == 'O'){                    que.push(Point(cur.v+addV[i], cur.h+addH[i]));                }            }        }    }    void solve(vector<vector<char>> &board) {        if(board.size() == 0 || board[0].size() == 0) return;        int W = board[0].size(), H = board.size();        queue<Point> que;        int i, j = 0;        for(i = 0; i < W; ++i){            if(board[0][i] == 'O') BFS(0, i, board, que);            if(H > 1 && board[H-1][i] == 'O') BFS(H-1, i, board, que); //遇到'O'，调用BFS        }        for(i = 0; i < H; ++i){            if(board[i][0] == 'O') BFS(i, 0, board, que);            if(W > 1 && board[i][W-1] == 'O') BFS(i, W-1, board, que);        }                for(i = 0; i < H; ++i){ //再次遍历全数组            for(j = 0; j < W; ++j){                if(board[i][j] == 'O') board[i][j] = 'X';                if(board[i][j] == 'Y') board[i][j] = 'O';            }        }    }private:    int addV[4] = {1, 0 , -1, 0};    int addH[4] = {0, 1, 0, -1};};

 

这样提交上去，超时。

 

有什么办法可以缩短时间呢？回想思路可以发现：每次遇到O，我们都调用一次BFS，每一次调用BFS，都需要清空队列que，然后再push。

为什么不把所有的BFS并为一次呢？每次遇到O，我们只向队列que中push当前O所在的Point，最后调用一次BFS集中处理que，其效果是完全一样的，但是时间上却省去了每次清空队列的时间。

代码：

class Solution {struct Point{    int h;    int v;    Point(int vp, int hp) : v(vp), h(hp) {};};public:    void solve(vector<vector<char>> &board) {        if(board.size() == 0 || board[0].size() == 0) return;        int W = board[0].size(), H = board.size();        queue<Point> que;        int i, j = 0;        for(i = 0; i < W; ++i){            if(board[0][i] == 'O') que.push(Point(0, i));            if(H > 1 && board[H-1][i] == 'O') que.push(Point(H-1, i)); //遇到O，只push，不再调用BFS        }        for(i = 0; i < H; ++i){            if(board[i][0] == 'O') que.push(Point(i, 0));            if(W > 1 && board[i][W-1] == 'O') que.push(Point(i, W-1)); //遇到O，只push，不再调用BFS        }        while(!que.empty()){ //调用BFS            Point cur = que.front();            que.pop();            board[cur.v][cur.h] = 'Y';            for(int i = 0; i < 4; i++){                if((cur.v+addV[i]) < H                && (cur.h+addH[i]) < W                && (cur.v+addV[i]) >= 0                && (cur.h+addH[i]) >= 0                 && board[cur.v+addV[i]][cur.h+addH[i]] == 'O'){                    que.push(Point(cur.v+addV[i], cur.h+addH[i]));                }            }        }                for(i = 0; i < H; ++i){            for(j = 0; j < W; ++j){                if(board[i][j] == 'O') board[i][j] = 'X';                if(board[i][j] == 'Y') board[i][j] = 'O';            }        }    }private:    int addV[4] = {1, 0 , -1, 0};    int addH[4] = {0, 1, 0, -1};};

 

 

小结：

这种需要更改Matrix的值的题目，上面解法用到了很简单的技巧：“引入一个中间值 Y”，避免了混淆。

 

例题 2

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

Could you devise a constant space solution?

class Solution {public:    void setZeroes(vector<vector<int> > &matrix) {            }};

 

这道题的困难之处在于“需要空间复杂度为常量”

有了上一题的启发，我们可以将0所在的行和列都设置为别的值X，最后将所有X设置为0。

但是这样做的弱点在于：

1. 每个元素都要被访问 >= 2次 (那些和0同行同列的元素被访问大于2次，那些不和0同行同列的元素被访问了2次)。

2. 如果题目改成vector<vector<bool> >，这种解法就失效了，因为没有第三个值可以引入。

 

如果空间复杂度要求是O(m+n)的话，我们会申明两个数组，分别来记录需要设为0的行号和列号。

进一步，如果如果空间复杂度要求是O(1)，我们虽然不能申明新数组，但是我们能用第一行和第一列来标记那些需要置为0的行和列。

当遇到 matix[i][j] == 0时，将matix[0][j] 和 matrix[i][0] 置为 0。

第一遍遍历matrix结束后，将所记录的行和列置为0。

这样做需要注意：如果第一行或者第一列有0，需要额外记录。

代码：

class Solution {public:    void setZeroes(vector<vector<int> > &matrix) {        if(matrix.size() == 0) return;        if(matrix[0].size() == 0) return;                bool firstRowSet = false;        bool firstColSet = false;        int i, j;        for(i = 0; i < matrix.size(); ++i){            for(j = 0; j < matrix[i].size(); ++j){                if(matrix[i][j] == 0){                    if(i == 0) firstRowSet = true;                    if(j == 0) firstColSet = true;                    matrix[i][0] = 0;                    matrix[0][j] = 0;                }            }        }                for(i = 1; i < matrix.size(); ++i){            if(matrix[i][0] == 0){                for(j = 1; j < matrix[i].size(); ++j)                    matrix[i][j] = 0;            }        }                for(j = 1; j < matrix[0].size(); ++j){            if(matrix[0][j] == 0){                for(i = 1; i < matrix.size(); ++i)                    matrix[i][j] = 0;            }        }                if(firstRowSet)            for(j = 1; j < matrix[0].size(); ++j)                matrix[0][j] = 0;        if(firstColSet)            for(i = 1; i < matrix.size(); ++i)                matrix[i][0] = 0;    }};