力扣top100 矩阵置零

fancy166166

于 2025-05-10 18:04:20 发布

阅读量241

点赞数 1

文章标签：矩阵人工智能算法

本文链接：https://blog.csdn.net/fancy166166/article/details/147856988

版权

开辟数组来标记元素为0的行和列，然后将对应的行和列的元素全部置为0；

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int n = matrix.size();
        int m = matrix[0].size();
        vector<int> l(m),r(n);
        for(int i = 0;i<n;i++){
            for(int j =0;j<m;j++){
                if(matrix[i][j]==0) l[j]=r[i]=1;
            }
        }
        for(int i = 0;i<n;i++){
            for(int j =0;j<m;j++){
                if(r[i]||l[j])
                    matrix[i][j]=0;
            }      
        }
    }
};

时间复杂度：因为用了两层for循环所以是O（mn）

空间复杂度：开辟了两个数组，是O（m+n）