LeetCode-73. 矩陣置零
阿新 • • 發佈:2021-03-23
給定一個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,然後直接原地更改即可 */ };