6 月更文挑战·零矩阵

89 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第4天,点击查看活动详情

一、题目描述:

编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。

示例 1:

输入: [ [1,1,1], [1,0,1], [1,1,1] ] 输出: [ [1,0,1], [0,0,0], [1,0,1] ] 示例 2:

输入: [ [0,1,2,0], [3,4,5,2], [1,3,1,5] ] 输出: [ [0,0,0,0], [0,4,5,0], [0,3,1,0] ]

来源:力扣(LeetCode) 链接:leetcode.cn/problems/ze… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二、思路分析:

我们可以用两个标记数组分别记录每一行和每一列是否有零出现。

具体地,我们首先遍历该数组一次,如果某个元素为 0,那么就将该元素所在的行和列所对应标记数组的位置置为 true。最后我们再次遍历该数组,用标记数组更新原数组即可。

步骤如下: 1、矩阵中某个数为零,则将该数所在行的第一个数置零,所在列的第一个数置零,即matrix0 = matrixi = 0 2、遍历上面处理过的首行和首列的每个元素(0位置除外),如果为零,则说明该行或者该列应该都为零,将该行或列全部元素置零

暴力解法判断

三、AC 代码:

class Solution {
    public void setZeroes(int[][] matrix) {
  int m=matrix.length;
        int n=matrix[0].length;
        boolean row0=false,cow0=false;
     
        for(int i=0;i<m;i++){//首列
            if(matrix[i][0]==0) cow0=true;
        }
        for(int j=0;j<n;j++){//首行
            if(matrix[0][j]==0) row0=true;
        }
        for(int i=1;i<m;i++){
            for(int j=1;j<n;j++){
                if(matrix[i][j]==0){
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }
        for(int i=1;i<m;i++){
            if(matrix[i][0]==0) matrix[i]=new int[n];
        }
        for(int j=1;j<n;j++){
            if(matrix[0][j]==0){
                for(int i=0;i<m;i++) matrix[i][j]=0;
            }
        }
        if(row0) matrix[0]=new int[n];
        if(cow0){
            for(int i=0;i<m;i++) matrix[i][0]=0;
        }
    }
}

四、总结:

今天这个题主要是分情况讨论,想明白各种情况,思路就清晰

掘友们,解题不易,留下个赞或评论再走吧!谢啦~ 💐

希望对你有帮助,期待您找到心意的工作和满意的offer

期待下次再见~

\