面试题 01.08. 零矩阵

2022-09-30 09:38:00 683 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

知识浅谈墨衍会员 · AI 创作全网分发关注 + 加入社区

软件工程：数据字典数据字典阅读详情

题意

编写一种算法，若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("]");
    }
}