LeetCode每日一题——矩阵置0
题目要求：
给定一个 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]]
代码实现：
void setZeroes(int** matrix, int matrixSize, int* matrixColSize) {
int m = matrixSize;//获取行数
int n = matrixColSize[0];//获取列数
int row[m],col[n];//用于标记0所在的行和列
memset(row, 0,sizeof(row));//初始化行数组中元素为0
memset(col, 0, sizeof(col));//初始化列数组中元素为0
for(int i = 0; i < m; i++)
{
for(int j = 0; j < n; j++)
{
if(!matrix[i][j])//!matrix[i][j]相当于matrix[i][j] == 0
{
row[i] = col[j] = true;//标记0所在的行和列
}
}
}
for(int i = 0; i < m; i++)
{
for(int j = 0; j < n; j++)
{
if(row[i] || col[j])//筛选出0所在的行和列
{
matrix[i][j] = 0;//让0所在行和列都为0
}
}
}
}
作者：力扣官方题解
来源：力扣（LeetCode）
坚持编程，我一直在路上！