LeetCode每日一题- day6

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;
            }
        }    
    }
};