lintcode-162-矩阵归零

162-矩阵归零

给定一个m×n矩阵，如果一个元素是0，则将其所在行和列全部元素变成0。

需要在原矩阵上完成操作。

样例

给出一个矩阵

[

[1, 2],

[0, 3]

]

返回

[

[0, 2],

[0, 0]

]

挑战

你是否使用了额外的空间？

• 一个直接的解决方案是使用 O(MN) 的额外空间，但这并不是一个好的方案。
• 一个简单的改进方案是使用 O(M + N) 的额外空间，但这仍然不是最好的解决方案。

你能想出一个常数空间的解决方案吗？

标签

Cracking The Coding Interview 矩阵

思路

若点 (i , j) 为0，则可以把 (i , 0) ,(0 , j) 标记为 0，最后根据该标记把列和行设置为 0，这样的设置会影响第 1 行和第 1 列，所以要先判断第 1 行和第 1 列是否包含 0，若包含 0，单独置第 1 行或第 1 列为0。

code

class Solution {
public:
    /*
     * @param matrix: An integer
     * @return:
     */

    void setZeroes(vector<vector<int>>& matrix) {
        // write your code here
        int sizeRow = matrix.size();
        if (sizeRow <= 0) {
            return;
        }
        int sizeCol = matrix[0].size(), i = 0, j = 0;
        bool firRow = false, firCol = false;
        // 检测第 1 行是否含0
        for (i = 0; i < sizeCol; i++) {
            if (matrix[0][i] == 0) {
                firRow = true;
                break;
            }
        }
        // 检测第 1 列是否含0
        for (i = 0; i < sizeRow; i++) {
            if (matrix[i][0] == 0) {
                firCol = true;
                break;
            }
        }
        // 若点 (i , j) 为0，则可以把 (i , 0)  ,(0 , j) 标记为 0
        for (i = 1; i < sizeRow; i++) {
            for (j = 1; j < sizeCol; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        // 根据 (i , 0)  ,(0 , j) 的值，单独设置某行或某列为 0
        for (i = 1; i < sizeCol; i++) {
            if (matrix[0][i] == 0) {
                for (j = 1; j < sizeRow; j++) {
                    matrix[j][i] = 0;
                }
            }
        }
        for (i = 1; i < sizeRow; i++) {
            if (matrix[i][0] == 0) {
                for (j = 1; j < sizeCol; j++) {
                    matrix[i][j] = 0;
                }
            }
        }
        // 第 1 行包含 0，置第 1 行全为 0
        if (firRow == true) {
            for (j = 0; j < sizeCol; j++) {
                matrix[0][j] = 0;
            }
        }
        // 第 1 列包含 0，置第 1 列全为 0
        if (firCol == true) {
            for (j = 0; j < sizeRow; j++) {
                matrix[j][0] = 0;
            }
        }
    }
};