LeetCode73-矩阵置零-数组-矩阵-模拟

原题链接

Note：

这里要注意的就是不能让更改操作影响后面的结果
就是不能这个新改成的0导致其他的行(列)变成0了
比如说第一个样例，不能说第一行第二个位置变成0了，结果后面操作的时候给第一行变成0了

要原地算法，直接用给的数组，第一行来记录这一列有没有0
第一列来记录每一行有没有0
开两个变量来记录第一行第一列有没有0

最后在遍历这些标志位来判断赋0即可

代码如下：

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int r0 = 1, c0 = 1;
        int n = matrix.size();
        if(!n)  return;
        int m = matrix[0].size();

        for(int i = 0; i < m; i ++)
            if(matrix[0][i] == 0){
                r0 = 0;
                break;
            }
        
        for(int i = 0; i < n; i ++)
            if(matrix[i][0] == 0){
                c0 = 0;
                break;
            }

        for(int i = 0; i < n; i ++)
            for(int j = 0; j < m; j ++){
                if(matrix[i][j] == 0)
                    matrix[0][j] = 0, matrix[i][0] = 0;
                }


        for(int i = 1; i < m; i ++)
            if(matrix[0][i] == 0)
                for(int j = 0; j < n; j ++)
                    matrix[j][i] = 0;
        
        for(int i = 1; i < n; i ++)
            if(matrix[i][0] == 0)
                for(int j = 0; j < m; j ++)
                    matrix[i][j] = 0;
        
        if(!r0)
            for(int i = 0; i < m; i ++)
                matrix[0][i] = 0;
        if(!c0)
            for(int i = 0; i < n; i ++)
                matrix[i][0] = 0;
        
    }
};