LeetCode (Set Matrix Zeroes)

矩阵零元素行列置零

胡为乎遑遑于何之

于 2017-05-12 19:34:04 发布

阅读量188

点赞数

CC 4.0 BY-SA版权

分类专栏： C++

本文链接：https://blog.youkuaiyun.com/m0_37705274/article/details/71756843

C++ 专栏收录该内容

83 篇文章

订阅专栏

本文介绍了一种高效的算法，用于处理给定的m x n矩阵。当矩阵中某个元素为0时，该算法能够将包含该元素的整行和整列的所有元素设置为0，并且操作是在原地进行的，无需额外的空间。

Problem:

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

Solution:

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size(), n = matrix[0].size();
        vector<int> row, col;
        int i = 0, j = 0;
        while(i < m){
            for(j = 0; j < n; j++){
                if(matrix[i][j] == 0){
                    row.push_back(i);
                    col.push_back(j);
                }
            }
            i++;
        }
        for(i = 0; i < row.size(); i++)
            for(j = 0; j < n; j++)
                matrix[row[i]][j] = 0;
        for(j = 0; j < col.size(); j++)
            for(i = 0; i < m; i++)
                matrix[i][col[j]] = 0;
    }
};