73. 矩阵置零

1.题目描述

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。
示例 1:

示例 2:

2.思路

1.每一行和每一列的第一个元素设置为标识位，如果为0，则标识该行或者该列存在0元素
2.因为第一行和第一列的元素都是matrix[0][0]，令其表示第一行，所以单独设置一个flag来标识第一列的情况
3.更新数值时，从右下角到左上角更新，因为标识位位于最左边和最上边，不能影响标识位的值

3.代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        bool flag = false;
        int m = matrix.size();
        int n = matrix[0].size();
        for(int i = 0;i < m;++i){
            if(matrix[i][0] == 0){//单独判断第一列
                flag = true;
            }
            for(int j = 1;j < n;++j){
                if(matrix[i][j] == 0){
                    matrix[0][j] = matrix[i][0] = 0;
                }
            }
        }
        for(int i = m - 1;i >= 0;--i){
            for(int j = n - 1;j >= 1;--j){
                if(matrix[i][0] == 0 || matrix[0][j] == 0){
                    matrix[i][j] = 0;
                }
            }
            if(flag){
                matrix[i][0] = 0;
            }
        }
    }
};

4.复杂度分析

时间复杂度：O(mn)
空间复杂度：O(1)