[leetbook中级算法-数组和字符串(java解法)]LC矩阵置零

leetbook:中级算法

类型:数组和字符串

题目名:矩阵置零

原题URL:https://leetcode-cn.com/leetbook/read/top-interview-questions-medium/xvmy42/

题目描述

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 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]
]

限制

• 一个直接的解决方案是使用 O(mn) 的额外空间，但这并不是一个好的解决方案。

• 一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。

• 你能想出一个常数空间的解决方案吗？

解题思路

1.根据题意,一下就能够想出来的方法就是使用m+n的空间,开一个row和col的数组来存放要变成全是0的行和列

2.也就是说,如果当前位置为0,那么就将当前位置的行和列都要变成0,存储在row和col的数组中

3.再次遍历这两个数组,将对应的matrix中的行和列全部置零.

4.那么常数空间呢?

5.这道题很显然至少需要依赖一个M+n的空间,如果没有空间,那么我们就从matrix里面借出来一个空间

6.用第一行第一列的空间借出来当空间

7.还是之前一样的做法,最后将第一行第一列看看要不要变成0就行

解题代码

class Solution {
    public void setZeroes(int[][] matrix) {
        if(matrix==null||matrix.length==0||matrix[0].length==0) return;
        //用第一行和第一列做标志位
        boolean row0 = false;
        boolean col0 = false;
        //int[] row = new int[matrix.length];
        //int[] col = new int[matrix[0].length];
        //如果第一行或者滴一列有一个是0.那么该行该列需要全部变成0
        for(int i = 0;i<matrix[0].length;i++) {
            if(matrix[0][i]==0) row0 = true;
        }
        for(int i = 0;i<matrix.length;i++) {
            if(matrix[i][0]==0) col0 = true;
        }
        for(int i = 1;i<matrix.length;i++) {
            for(int j = 1;j<matrix[0].length;j++) {
                if(matrix[i][j]==0) {
                    matrix[0][j]=0;
                    matrix[i][0]=0;
                }
            }
        }
        for(int i = 1;i<matrix.length;i++) {
            for(int j = 1;j<matrix[0].length;j++) {
                if(matrix[i][0]==0||matrix[0][j]==0) matrix[i][j] = 0;
            }
        }
        if(row0) {
            for(int i = 0;i<matrix[0].length;i++) {
                matrix[0][i]=0;
            }
        }
        if(col0) {
            for(int i = 0;i<matrix.length;i++) {
                matrix[i][0]=0;
            }
        }
    }
}