LeetCode刷题——73.矩阵置零

面试刷题第六天

LeetCode刷题——73.矩阵置零

1、题目描述

给定一个m x n的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。

2、解题思路

（1）利用两个布尔类型的变量，记录第一行和第一列是否有0。
（2）遍历除第一行第一列以外的矩阵元素，如果有元素为0，则将对应的第一行第一列的位置置为0。
（3）同样遍历除第一行第一列以外的矩阵元素，如果matrix[i][j]所在位置的第一行、第一列有任意一个位置为0，那么那个位置将被置为0。
（4）根据两个布尔类型的变量，决定是否将第一行、第一列置为0。
Java代码如下：

class Solution {
    public void setZeroes(int[][] matrix) {
        boolean row0Zero = false;
        boolean col0zero = false;
        int i = 0;
        int j = 0;
        //判断第0行有没有0
        for(i=0; i<matrix[0].length; i++){
            if(matrix[0][i] == 0){
                row0Zero = true;
                break;
            }
        }
        //判断第0列有没有0
        for(i=0; i<matrix.length; i++){
            if(matrix[i][0] == 0){
                col0zero = true;
                break;
            }
        }
        //中间部分
        for(i=1; i<matrix.length; i++){
            for(j=1; j<matrix[0].length; j++){
                if(matrix[i][j] == 0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        //先更新中间部分
        for(i=1; i<matrix.length; i++){
            for(j=1; j<matrix[0].length; j++){
                if(matrix[i][0] == 0 || matrix[0][j] == 0){
                    matrix[i][j] = 0;
                }
            }
        }
        //更新第0行
        if(row0Zero){
            for(i=0; i<matrix[0].length; i++){
                matrix[0][i] = 0;
            }
        }
        //更新第0列
        if(col0zero){
            for(i=0; i<matrix.length; i++){
                matrix[i][0] = 0;
            }
        }
    }
}