每日一题——矩阵置零

菜鸡每日一题系列打卡73天

每天一道算法题目 

小伙伴们一起留言打卡

坚持就是胜利，我们一起努力！

题目描述(引自LeetCode)

给定一个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)。这意味着我们不能靠额外的数组记录每个0出现的位置。

为了实现原地算法，我们可以选取某一行为标志行，某一列为标志列，方便起见，我们选择第一行为标志行，第一列为标志列。这样我们就可以利用第一行和第一列中的元素来记录本列和本行是否需要置零。

值得注意的是，我们选取的标志行和标志列需要进行额外的处理。

话不多说，上代码！

代码实现

class Solution {


    public void setZeroes(int[][] matrix) {
        // 特殊情况处理
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0) return;


        int row = matrix.length;
        int column = matrix[0].length;


        // 设置第一行、第一列标志
        // 用于判断第一行和第一列是否需要全部置零
        boolean rFlag = false, cFlag = false;
        for (int j = 0; j < column; j++) {
            if (matrix[0][j] == 0) rFlag = true;
        }
        for (int i = 0; i < row; i++) {
            if (matrix[i][0] == 0) cFlag = true;
        }


        // 更新每行和每列的第一个数字
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < column; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }


        // 更新除第一行、第一列以外的整个矩阵
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < column; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) matrix[i][j] = 0;
            }
        }


        // 判断第一行是否需要更新，若是则更新
        if (rFlag) {
            for (int j = 0; j < column; j++) {
                matrix[0][j] = 0;
            }
        }


        // 判断第一列是否需要更新，若是则更新
        if (cFlag) {
            for (int i = 0; i < row; i++) {
                matrix[i][0] = 0;
            }
        }
    }


}

代码分析

对代码进行分析，程序主要的时间消耗在对矩阵的两次遍历上，因此，时间复杂度为O(mn)；而就空间而言，仅仅使用了常数级别的额外空间，因此，空间复杂度为O(1)。

执行结果

学习 | 工作 | 分享

????长按关注“有理想的菜鸡”

只有你想不到，没有你学不到