LeetCode 73.矩阵置零

题目：

给定一个 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
• -231 <= matrix[i][j] <= 231 - 1

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/set-matrix-zeroes

题解：

使用两个数组用来标记每一行和每一列是否有0出现。遍历原数组，如果某个元素为0，就将该元素所在行和列对应的标记数组的行列位置标为 true ；再遍历一次该数组，对照行列标记数组进行更新。

void setZeroes(int** matrix, int matrixSize, int* matrixColSize){
    int m=matrixSize;
    int n=matrixColSize[0];
    int row[m],col[n];
    memset(row,0,sizeof(row));
    memset(col,0,sizeof(col));
    for(int i=0;i<m;i++){
        for(int j=0;j<n;j++){
            if(matrix[i][j]==0) row[i]=col[j]=true;
        }
    }
    for(int i=0;i<m;i++){
        for(int j=0;j<n;j++){
            if(row[i]||col[j]) matrix[i][j]=0;
        }
    }
}

运行结果：