Set Matrix Zeroes

矩阵中元素为零时设置整行整列为零的高效算法实现

最新推荐文章于 2020-02-14 21:39:11 发布

原创最新推荐文章于 2020-02-14 21:39:11 发布 · 465 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

636 篇文章

订阅专栏

本文详细介绍了如何使用C++编程语言实现一个矩阵操作函数，该函数能够在矩阵中找到指定元素为零的位置，并将与其对应的整个行和列设置为零，同时保持其他部分的原始状态。通过引入集合数据结构来记录零元素的位置，该算法实现了在原地操作矩阵的目标，避免了额外内存的使用。

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) {
        set<int> row;
	    set<int> column;
	    int m = matrix.size();
	    for (int i = 0; i < m; i++)
	    {
		    int n = matrix[i].size();
		    for (int j = 0; j < n; j++)
		    {
			    if (matrix[i][j] == 0)
			    {
				    row.insert(i);
				    column.insert(j);
			    }
		    }
	    }

	    for (set<int>::iterator it = row.begin(); it != row.end(); it++)
	    {
		    int size = matrix[*it].size();
		    for (int j = 0; j < size; j++)
		    {
			    matrix[*it][j] = 0;
		    }
	    }

	    for (set<int>::iterator it = column.begin(); it != column.end(); it++)
	    {
		    for (int j = 0; j < m; j++)
		    {
		    	matrix[j][*it] = 0;
		    }
	    }
    }
};