signed

QiShunwang

“诚信为本、客户至上”

LeetCode-73. 矩阵置零

2021/3/21 10:53:04   来源:

给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

进阶:

一个直观的解决方案是使用  O(mn) 的额外空间,但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗?
 

示例 1:


输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出:[[1,0,1],[0,0,0],[1,0,1]]

示例 2:


输入:matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出:[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

提示:

m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-231 <= matrix[i][j] <= 231 - 1

 

我的思路:

存储键值对,(有重复的键,用multimap!)然后直接遍历修改。

#include <iostream>
#include <vector>
#include <map>
using namespace std;

/* 使用multimap,记录0炸弹(泡泡堂) */
class Solution {
public:
	void setZeroes(vector<vector<int>>& matrix) {
		for (int i = 0; i < matrix.size(); i++) {
			for (int j = 0; j < matrix[i].size(); j++) {
				if (matrix[i][j] == 0) {
					zeroMap.insert(make_pair(i, j));
				}
			}
		}

		multimap<int, int>::iterator it = zeroMap.begin();
		while (it != zeroMap.end()) {
			setMatrixZeroX(matrix, it->first);
			auto second = zeroMap.equal_range(it->first);
			for (auto jt = second.first; jt != second.second; jt++) {
				setMatrixZeroY(matrix, jt->second);
			}
			it++;
		}
	}

public:
	/* 相同的行数,更改为0 */
	void setMatrixZeroX(vector<vector<int>>& matrix, int x) {
		for (int j = 0; j < matrix[x].size(); j++) {
			matrix[x][j] = 0;
		}
	}
	/* 不同的列数,更改为0 */
	void setMatrixZeroY(vector<vector<int>>& matrix, int y) {
		for (int i = 0; i < matrix.size(); i++) {
			matrix[i][y] = 0;
		}
	}

private:
	multimap<int, int> zeroMap; /* 记录相同键值的value,然后直接原地更改即可 */
};