LeetCode_Set Matrix Zeros

最新推荐文章于 2024-08-16 09:55:14 发布

原创最新推荐文章于 2024-08-16 09:55:14 发布 · 750 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一种高效的算法，用于处理给定的m x n矩阵：当矩阵中某个元素为0时，将该元素所在行和列的所有元素都设置为0。此操作在原地完成，无需额外空间。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

class Solution {
public:
    void setZeroes(vector<vector<int> > &matrix) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        int row = matrix.size();
        int col = matrix[0].size();
        vector<bool> rowVec;
        rowVec.resize(row, false);
        vector<bool> colVec;
        colVec.resize(col, false);
        
        for (int i = 0; i < row; ++i)
        {
            for (int j = 0; j < col; ++j)
            {
                if (matrix[i][j] == 0)
                {
                    rowVec[i] = true;
                    colVec[j] = true;
                }
            }
        }
        for (int i = 0; i < row; ++i)
        {
            if (rowVec[i])
            {
                for (int j = 0; j < col; ++j)
                {
                    matrix[i][j] = 0;
                }
            }
        }
        for (int j = 0; j < col; ++j)
        {
            if (colVec[j])
            {
                for (int i = 0; i < row; ++i)
                {
                    matrix[i][j] = 0;
                }
            }
            
        }
    }
};