面试题 01.08. 零矩阵

知识浅谈 🐄🐄🐄🐄🐄大佬
后端领域优质创作者
博客专家认证
2022-09-30 09:38:00

img

题意

编写一种算法,若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]
]

题解

row col分别记录是否有第一行为0的位置,因为其他节点为0的话,把其他节点对应的行和列的第一个位置置为0.

class Solution {
    public void setZeroes(int[][] matrix) {
        boolean row=false,col=false;
        int m=matrix.length,n=matrix[0].length;
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(matrix[i][j]==0){
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                    if(i==0) row=true; 
                    if(j==0) col=true;
                }
            }
        }
        for(int i=1;i<m;i++)
            if(matrix[i][0]==0)
                for(int j=0;j<n;j++) matrix[i][j]=0;
        for(int i=1;i<n;i++)
            if(matrix[0][i]==0)
                for(int j=0;j<m;j++) matrix[j][i]=0;

        if(row)
            for(int i=0;i<n;i++) matrix[0][i]=0;
        if(col)
            for(int i=0;i<m;i++) matrix[i][0]=0;
        // System.out.println("[");
        // for(int i=0;i<m;i++){
        //     System.out.print("[");
        //     for(int j=0;j<n;j++){
        //         if(j!=n-1) System.out.print(matrix[i][j]+",");
        //         else System.out.print(matrix[i][j]);
        //     }
        //     if(i!=m-1) System.out.println("],");
        //     else System.out.println("]");
        // }
        // System.out.println("]");
    }
}

img

...全文
17 回复 打赏 收藏 举报
写回复
回复
切换为时间正序
请发表友善的回复…
发表回复
发帖
知识浅谈后端社区

3998

社区成员

知识浅谈(带你学后端)社区,旨在为学习后端的小伙伴提供技术内容共享学习平台,助力每位小伙伴写出高质量代码,欢迎各位小伙伴的加入。
java爬虫后端 个人社区 吉林省·长春市
社区管理员
  • 知识浅谈
加入社区
帖子事件
创建了帖子
2022-09-30 09:38
社区公告

【社区倡导与禁止】

  • 社区倡导积极发帖,互相交流技术问题,学习心得。
  • 社区不限制技术领域,后端内容增加加精权重
  • 社区禁止发布低俗、不健康的内容

【社区积分规则】

  • 在社区「发帖」得10积分
  • 内容被管理员「加精」得10积分
  • 点赞他人内容得1积分
  • 评论内容得2积分