【算法训练】LeetCode#73 矩阵置零.md

一、描述

73. 矩阵置零

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

示例 1：

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

示例 2：

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

二、思路

递归解决呢，每个0元素的上下左右做递归。（哦，理解错题意了，我以为手动置0的位置也会影响周围呢）

因为有空间限制，只能用O(1)空间复杂度的算法，因此可以循环时记录下0的位置，再遍历一次做赋值操作。

三、解题

public class LeetCode73 {
    public void setZeroes(int[][] matrix) {
        LinkedList<int[]> arr = new LinkedList<>();
        if (matrix == null){
            return;
        }
        int m = matrix.length;
        int n = matrix[0].length;
        for (int i = 0 ; i < matrix.length ; i++){
            for (int j = 0 ; j < matrix[0].length ; j++){
                if (matrix[i][j] == 0){
                    arr.add(new int[]{i,j}); // 记录0的位置
                }
            }
        }
        for (int[] loc : arr){
            // 上下
            for (int i = 0 ; i < m ; ++i){
                // 列不变，行变
                matrix[i][loc[1]] = 0;
            }

            // 左右
            for (int i = 0 ; i < n ; ++i){
                // 行不变，列变
                matrix[loc[0]][i] = 0;
            }
        }
    }
}