【LeetCode刷题记录】73. 矩阵置零

最新推荐文章于 2024-10-18 10:39:20 发布

C_greenbird

最新推荐文章于 2024-10-18 10:39:20 发布

阅读量648

点赞数 8

分类专栏： LeetCode 文章标签： leetcode 矩阵算法 c++

本文链接：https://blog.csdn.net/C_greenbird/article/details/137783775

版权

LeetCode 专栏收录该内容

51 篇文章 0 订阅

订阅专栏

文章讲述了如何使用原地算法解决矩阵置零问题，通过两次遍历，首先标记行中零元素的索引，然后将这些行和对应的列置零。代码展示了如何在C++中实现这一过程。

摘要由CSDN通过智能技术生成

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]]

提示：
$m == ma t r i x . l e n g t h$
$n == ma t r i x [0] . l e n g t h$
$1 <= m, n <= 200$
$2^{31} <= matrix[i][j] <= 2^{31} - 1$

思路

本题采用标记+遍历两次即可。
首先按行遍历，查找矩阵每行的零元素，记录零元素的索引号，并将该行置0，采用fill函数。
之后再次按行遍历时，用记录的索引号将零元素所在列置0。

代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        vector<int> zero;
        for (int i = 0; i < matrix.size(); i++) {
            int tmp = -1;
            for (int j = 0; j < matrix[i].size(); j++) {
                if (matrix[i][j] == 0) {
                    zero.push_back(j);
                    tmp = i;
                }
            }
            // 将0所在行置0
            if (tmp != -1) {
                fill(matrix[i].begin(), matrix[i].end(), 0);
            }
        }
        for (int i = 0; i < matrix.size(); i++) {
        	// 将0所在列置0
            for (int j = 0; j < zero.size(); j++) {
                matrix[i][zero[j]] = 0;
            }
        }
    }
};