【LeetCode面试150】——73矩阵置零

博客昵称：沈小农学编程

作者简介：一名在读硕士，定期更新相关算法面试题，欢迎关注小弟！

PS：哈喽！各位CSDN的uu们，我是你的小弟沈小农，希望我的文章能帮助到你。欢迎大家在评论区唠嗑指正，觉得好的话别忘了一键三连哦！😘

题目难度：中等

默认优化目标：最小化时间复杂度。

Python默认为Python3。

目录

1 题目描述

2 题目解析

3 算法原理及代码实现

3.1 使用两个标记变量

参考文献

---

1 题目描述

给定一个 *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]]

提示：

• m == matrix.length

• n == matrix[0].length

• 1 <= m, n <= 200

• -231 <= matrix[i][j] <= 231 - 1

进阶：

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

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

• 你能想出一个仅使用常量空间的解决方案吗？

2 题目解析

输入是一个m×n的二维矩阵matrix，输出是一个m×n的二维矩阵ans。约束条件是把matrix中值为零的元素所在行和列的元素值变为零。

3 算法原理及代码实现

3.1 使用两个标记变量

不使用辅助数组，原地将满足条件的元素变为零。难度在于找到matrix中原始值为零的元素后，将其所在行和列变为零后，后破坏原来的matrix中为零值的元素。

我们可以将matrix的第一行和第一列作为标记位置。如果matrix[i][j]=0，把matrix[i][0]和matrix[0][j]变为0，之后根据标志位将对应行与列的元素变为零。

因为第一行与第一列用于标志，所以我们需要两个变量来记录第一行与第一列中是否有元素为0，有在把第一行或者第一列变为0，无则不改动。

流程图如下所示：

时间复杂度为O(mn)，空间复杂度为O(1)。

C++代码实现

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m=matrix.size(),n=matrix[0].size();
        bool row=false,column=false;
        for(int i=0;i<m;i++){
            if(!matrix[i][0]){
                column=true;
                break;
            }
        }
        for(int j=0;j<n;j++){
            if(!matrix[0][j]){
                row=true;
                break;
            }
        }
        for(int i=1;i<m;i++){
            for(int j=1;j<n;j++){
                if(!matrix[i][j]){
                    matrix[i][0]=matrix[0][j]=0;
                }
            }
        }
        for(int i=1;i<m;i++){
            for(int j=1;j<n;j++){
                if(!matrix[i][0] || !matrix[0][j]){
                    matrix[i][j]=0;
                }
            }
        }
        if(column){
            for(int i=0;i<m;i++){
                matrix[i][0]=0;
            }
        }
        if(row){
            for(int j=0;j<n;j++){
                matrix[0][j]=0;
            }
        }
    }
};

Python代码实现

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        m, n = len(matrix), len(matrix[0])
        row, column = False, False
        
        for i in range(m):
            if matrix[i][0] == 0:
                column = True
                break
        
        for j in range(n):
            if matrix[0][j] == 0:
                row = True
                break
        
        for i in range(1, m):
            for j in range(1, n):
                if matrix[i][j] == 0:
                    matrix[i][0] = matrix[0][j] = 0
        
        for i in range(1, m):
            for j in range(1, n):
                if matrix[i][0] == 0 or matrix[0][j] == 0:
                    matrix[i][j] = 0
        
        if column:
            for i in range(m):
                matrix[i][0] = 0
        
        if row:
            for j in range(n):
                matrix[0][j] = 0

​

参考文献

力扣面试经典150题

力扣官方题解