【leetcode】73. Set Matrix Zeroes

使用二维vector高效置零矩阵元素

最新推荐文章于 2025-11-05 23:47:48 发布

原创最新推荐文章于 2025-11-05 23:47:48 发布 · 449 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #matrix

【leetcode】专栏收录该内容

53 篇文章

订阅专栏

本文介绍了一种利用二维vector快速定位并置零矩阵中所有为零元素的方法，同时通过两个bool数组来记录行和列的清零状态，避免了重复操作，提升了效率。

开个二维vector进行保存0所在的位置，然后通过便利存储的位置置0。

注意，同时开两个bool数组，来保存行列的清零情况，以免重复清零。

<pre name="code" class="cpp">/**
 *  @author         johnsondu
 *  @time           11:26 14th Oct 2015
 *  @type           Simple problem..
 *  @status         Accepted
 *  @problem        Set Matrix Zeroes
 *  @url            https://leetcode.com/problems/set-matrix-zeroes/
 */

 class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        vector<vector<int>> pos;
        int row = matrix.size();
        if(row == 0) return;
        int col = matrix[0].size();
        for(int i = 0; i < row; i ++) 
            for(int j = 0; j < col; j ++) 
                if(matrix[i][j] == 0) {
                    pos.push_back({i, j});
                }
        vector<bool> vecRow(row, false);
        vector<bool> vecCol(col, false);
        for(int i = 0; i < pos.size(); i ++) {
            int x = pos[i][0];
            int y = pos[i][1];
            if(!vecRow[x])
            {
                for(int i = 0; i < col; i ++) matrix[x][i] = 0;
                vecRow[x] = true;
            }
                if(!vecCol[y]) {
                for(int i = 0; i < row; i ++) matrix[i][y] = 0;
                vecCol[y] = true;
            }
        }
    }
};