c++实现矩阵置零

在计算机科学中，矩阵是一个非常重要的数据结构。在实际应用中，我们经常需要对矩阵进行一些操作。本篇文章将介绍如何使用C++实现矩阵置零的算法。

问题描述

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

解决方案

我们可以使用两个数组来记录哪些行和列需要被置零。首先，我们遍历整个矩阵，如果一个元素为0，我们就将其所在的行和列分别标记为需要被置零的行和列。然后，我们再次遍历整个矩阵，如果一个元素所在的行或列被标记为需要被置零的行或列，我们就将该元素置零。

下面是C++代码实现：

void setZeroes(vector<vector<int>>& matrix) {
    int m = matrix.size();
    int n = matrix[0].size();
    vector<int> rows(m, 0);
    vector<int> cols(n, 0);
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (matrix[i][j] == 0) {
                rows[i] = 1;
                cols[j] = 1;
            }
        }
    }
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (rows[i] == 1 || cols[j] == 1) {
                matrix[i][j] = 0;
            }
        }
    }
}

时间复杂度：O(mn)，其中m和n分别是矩阵的行数和列数。我们需要遍历整个矩阵两次。

空间复杂度：O(m+n)，我们需要两个大小分别为m和n的数组来记录哪些行和列需要被置零。

总结

本篇文章介绍了如何使用C++实现矩阵置零的算法。该算法的时间复杂度为O(mn)，空间复杂度为O(m+n)。如果您有任何问题或建议，请在评论区留言，谢谢！