1. 程式人生 > 其它 >LeetCode-73. 矩陣置零

LeetCode-73. 矩陣置零

給定一個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,然後直接原地更改即可 */
};