LeetCode每日一题- day6
新手入坑LeetCode,每天打卡一道题
算法不一定很好,只是我自己的一个水平体现,做个自己刷题的记录,欢迎交流学习
(尽量AC LeetCode官方的每日一题)
欢迎交流学习!
题目:面试题 01.08. 零矩阵
编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。
样例:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
思路:
使用一个新的二位数组,记录下每个0
元素的位置{i,j}
然后遍历,如果有0
,则使该i行和j列
的元素全部更新为0
代码:
class Solution {
public:
const static int N = 201,M = 201;
int p[N][M];
void setZeroes(vector<vector<int>>& matrix) {
if(matrix.size() == 0 || matrix[0].size() == 0) return ;
//定义一个二维数组,记录元素0的位置[i][j]
//然后两次for循环,把i行和j列全部修改为0
int n = matrix.size(), m = matrix[0].size();
for(int i = 0; i < n; i ++){
for(int j = 0; j < m; j ++){
if(matrix[i][j] == 0){
//p[i][j] = 1;//最初的定位0元素的想法
//但实际可以把i行和j列同时修改为1
for(int k = 0; k < n; k ++) p[k][j] = 1;
for(int k = 0; k < m; k ++) p[i][k] = 1;
}
}
}
for(int i = 0; i < n; i ++){
for(int j = 0; j < m; j ++){
if(p[i][j] == 1) matrix[i][j] = 0;
}
}
}
};