519. 隨機(jī)翻轉(zhuǎn)矩陣

給你一個 m x n 的二元矩陣 matrix ，且所有值被初始化為 0 。請你設(shè)計一個算法，隨機(jī)選取一個滿足?matrix[i][j] == 0 的下標(biāo)?(i, j) ，并將它的值變?yōu)?1 。所有滿足 matrix[i][j] == 0 的下標(biāo) (i, j) 被選取的概率應(yīng)當(dāng)均等。

盡量最少調(diào)用內(nèi)置的隨機(jī)函數(shù)，并且優(yōu)化時間和空間復(fù)雜度。

實現(xiàn) Solution 類：

• Solution(int m, int n) 使用二元矩陣的大小 m 和 n 初始化該對象
• int[] flip() 返回一個滿足?matrix[i][j] == 0 的隨機(jī)下標(biāo) [i, j] ，并將其對應(yīng)格子中的值變?yōu)?1
• void reset() 將矩陣中所有的值重置為 0

示例：輸入 ["Solution", "flip", "flip", "flip", "reset", "flip"] [[3, 1], [], [], [], [], []] 輸出 [null, [1, 0], [2, 0], [0, 0], null, [2, 0]]解釋 Solution solution = new Solution(3, 1); solution.flip(); // 返回 [1, 0]，此時返回 [0,0]、[1,0] 和 [2,0] 的概率應(yīng)當(dāng)相同 solution.flip(); // 返回 [2, 0]，因為 [1,0] 已經(jīng)返回過了，此時返回 [2,0] 和 [0,0] 的概率應(yīng)當(dāng)相同 solution.flip(); // 返回 [0, 0]，根據(jù)前面已經(jīng)返回過的下標(biāo)，此時只能返回 [0,0] solution.reset(); // 所有值都重置為 0 ，并可以再次選擇下標(biāo)返回 solution.flip(); // 返回 [2, 0]，此時返回 [0,0]、[1,0] 和 [2,0] 的概率應(yīng)當(dāng)相同

提示：

• 1 <= m, n <= 104
• 每次調(diào)用flip 時，矩陣中至少存在一個值為 0 的格子。
• 最多調(diào)用 1000 次 flip 和 reset 方法。

解題思路

對于矩陣的每個位置都使用編號代替，編號范圍為[0,n*m-1]，使用set存儲已經(jīng)被標(biāo)記為1的位置，每次我們生成一個該區(qū)間內(nèi)的隨機(jī)數(shù)，檢查該編號對應(yīng)的位置是否為1，如果為1的話，則重新生成隨機(jī)數(shù)，否則返回該隨機(jī)數(shù)對應(yīng)矩陣位置的下標(biāo)，并且加入set中。

代碼

class Solution { public:int mm,nn;unordered_set<int> set;Solution(int m, int n) {this->mm=m;this->nn=n;}vector<int> flip() {int cur=rand()%(nn*mm);while (set.count(cur)>0){cur=rand()%(nn*mm);}set.insert(cur);return {cur/nn,cur%nn};}void reset() {set.clear();} }; /*** Your Solution object will be instantiated and called as such:* Solution* obj = new Solution(m, n);* vector<int> param_1 = obj->flip();* obj->reset();*/

總結(jié)

以上是生活随笔為你收集整理的519. 随机翻转矩阵的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。