数组与字符串——零矩阵

没时间了，先上题！
编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

解题思路

• 低效思路：在利用双重循环遍历矩阵的同时只要碰到为0的数就把该数所在的行和列全部修改为0，得到的时间复杂度很高。

• 高效思路：采用标记行列数组的方式。首先利用vector的一种创建确定元素初始值和个数的方式创建两个容器，即vector< int > vr(row, false)，vector< int > vc(column, false)。然后遍历矩阵，在遍历的同时只要碰到0就把vr中对应行下标的元素值标为true，对vc也执行相同操作。循环结束后，再次遍历整个数组，只要碰到对应的vc或vr中值为true的便将矩阵中的元素值修改为0。

代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int row = matrix.size();
        int column = matrix[0].size();
        if(row == 0 || column == 0) {
            return;
        }
        //vector的一种初始化方式：第一个参数为vector中的元素个数，第二个参数为指定初始化值
        vector<int> vr(row, false);
        vector<int> vc(column, false);
        
        for(int i = 0; i < row; i++) {
            for(int j = 0; j < column; j++) {
                if(matrix[i][j] == 0) {
                    vr[i] = true;
                    vc[j] = true;
                }
            }
        }
        
        for(int i = 0; i < row; i++) {
            for(int j = 0; j < column; j++) {
                if(vr[i] || vc[j]) {
                    matrix[i][j] = 0;
                }
            }
        }
    }
};

提交记录