73、矩阵置零

1）题目分析

给定一个 *m* x *n* 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。

示例 1：

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

示例 2：

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

提示：

• m == matrix.length
• n == matrix[0].length
• 1 <= m, n <= 200
• -2^31 <= matrix[i][j] <= 2^31 - 1

2）分析

• 用两个变量来标记第一行和第一列是否有零；
• 用第一行来标记每一列是否有零；
• 用第一列来标记每一行是否有零；
• 用第一行和第一列来更新矩阵；
• 用两个标记变量来更新第一行和第一列。

3）C++代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int row=matrix.size();
        int col=matrix[0].size();

        bool col_0=false;
        bool row_0=false;

        //判断第一列是否有零
        for(int i=0;i<row;i++){
            if(!matrix[i][0]){
                col_0=true;
                break;
            }
        }
        //判断第一行是否有零
        for(int i=0;i<col;i++){
            if(!matrix[0][i]){
                row_0=true;
                break;
            }
        }

        //使用第一行第一列来进行记录每一行每一列是否有零
        for(int i=1;i<row;i++){
            for(int j=1;j<col;j++){
                if(!matrix[i][j]){
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }

        //使用第一行第一列来更新矩阵
        for(int i=1;i<row;i++){
            for(int j=1;j<col;j++){
                if(!matrix[i][0]||!matrix[0][j])
                    matrix[i][j]=0;
            }
        }

        //使用col_0和row_0来更新第一行和第一列
        if(col_0){
            for(int i=0;i<row;i++)
                matrix[i][0]=0;
        }
        if(row_0){
            for(int i=0;i<col;i++)
                matrix[0][i]=0;
        }
    }
};