题目:
给定一个 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;
}
}
}
运行结果: