原地算法-LeetCode73-矩阵置零

题目

给定一个 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) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个常数空间的解决方案吗？

原地算法

原地算法一般来说只能额外使用常数空间O(1),基本思路是利用输入的空间而不是重新创建新的空间。

思路1

创建一个新的m*n的矩阵。额外空间O(mn)。

思路2

创建两个数组a[m]和a[n],分别记录矩阵行和列是否有0。额外空间O(m+n)。

思路3

在输入矩阵中找到一个0的行和列（替代思路2中的a[m]和a[n]）,分别记录矩阵行和列是否有0。额外空间O(1)。

代码

class Solution {
    public void setZeroes(int[][] matrix) {
        //找到一个0的行和列,用于记录需要清0的列和行
        int hang=-1;
        int lie=-1;
        for(int i=0;i<matrix.length;i++){
            for(int j=0;j<matrix[0].length;j++){
                if(matrix[i][j]==0){
                    hang=i;
                    lie=j;
                    break;
                }
            }
            if(hang!=-1){
                break;
            }
        }
        if(hang==-1){
            return;
        }
        //用上面找到的这个0的行和列，分别记录需要清0的列和行（置0）
        for(int i=0;i<matrix.length;i++){
            if(i==hang){
                continue;
            }
            for(int j=0;j<matrix[0].length;j++){
                if(j==lie){
                    continue;
                }
                if(matrix[i][j]==0){
                    matrix[i][lie]=0;
                    matrix[hang][j]=0;
                }
            }
        }
        //清0
        for(int i=0;i<matrix[0].length;i++){
            if(matrix[hang][i]==0&&i!=lie){
                for(int j=0;j<matrix.length;j++){
                    matrix[j][i]=0;
                }
            }
        }
        for(int i=0;i<matrix.length;i++){
            if(matrix[i][lie]==0&&i!=hang){
                for(int j=0;j<matrix[0].length;j++){
                    matrix[i][j]=0;
                }
            }
        }
        //用于记录的行和列也需要清0
        for(int i=0;i<matrix.length;i++){
            matrix[i][lie]=0;
        }
        for(int i=0;i<matrix[0].length;i++){
            matrix[hang][i]=0;
        }
    }
}

总结

原地算法一般来说只能额外使用常数空间O(1),基本思路是利用输入的空间而不是重新创建新的空间。