力扣 题目73- 矩阵置零

题目

题解

其实思考一下就知道 我们必须记录一下遍历时0的xy 然后根据记录再去放0

但是如果我们使用其他容器去记录那么我们遍历完还要遍历 记录容器 时间复杂度肯定会搞

有没有一种方法可以在当前容器记录呢?

我们使用第一行与第一列记录 如果

if (matrix[i][j] == 0) {
matrix[0][j] = 0;
matrix[i][0] = 0;
}

这样我们只要重新遍历第一列与第一行 根据0再放对行列至0即可

但是我们发现第一列与第一行的处理有点问题 遍历会导致变0 然后导致其他全为0     所以我们需要提前遍历 看看是否有0 特殊处理 如果有则最后再将第一列或第一行放0

代码

#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        bool rowFlag = false;
        bool colFlag = false;
        //看看第一列行否有0
        for (int j = 0; j < matrix[0].size(); j++) {
            if (matrix[0][j] == 0) {
                rowFlag = true;
                break;
            }
        }
        //看看第一列是否有0
        for (int i = 0; i < matrix.size(); i++) {
            if (matrix[i][0] == 0) {
                colFlag = true;
                break;
            }
        }

        for (int i = 1; i < matrix.size(); i++) {
            for (int j = 1; j < matrix[i].size(); j++) {
                if (matrix[i][j] == 0) {
                    matrix[0][j] = 0;
                    matrix[i][0] = 0;
                }
            }
        }
        //遍历看对应的第一列是否为0
        for (int i = 1; i < matrix.size(); i++) {
            if (matrix[i][0] == 0) {
                for (int j = 0; j < matrix[i].size(); j++) {
                    matrix[i][j] = 0;
                }
            }
        }
        //遍历看对应的第一行是否为0
        for (int j = 1; j < matrix[0].size(); j++) {
            if (matrix[0][j] == 0) {
                for (int i = 0; i < matrix.size(); i++) {
                    matrix[i][j] = 0;
                }
            }
        }
        //特殊处理第一列
        if (colFlag) {
            for (int i = 0; i < matrix.size(); i++) {
                matrix[i][0] = 0;
            }
        }

        //特殊处理第一行
        if (rowFlag) {
            for (int j = 0; j < matrix[0].size(); j++) {
                matrix[0][j] = 0;
            }
        }
       

        
    }
};

int main() {
    Solution sol;
    vector<vector<int>> matrix = { {0, 1, 2, 0},{3, 4, 5, 2},{1, 3, 1, 5} };
    sol.setZeroes(matrix);
    for (int i = 0; i < matrix.size(); i++) {
        for (int j = 0; j < matrix[i].size(); j++) {
            cout << matrix[i][j]<<" ";
        }
        cout << endl;
    }

}