Leetcode:519. 随机翻转矩阵

题中给出一个 n 行 n 列的二维矩阵 (n_rows,n_cols)，且所有值被初始化为 0。要求编写一个 flip 函数，均匀随机的将矩阵中的 0 变为 1，并返回该值的位置下标 [row_id,col_id]；同样编写一个 reset 函数，将所有的值都重新置为 0。尽量最少调用随机函数 Math.random()，并且优化时间和空间复杂度。

注意:

1.1 <= n_rows, n_cols <= 10000

2. 0 <= row.id < n_rows 并且 0 <= col.id < n_cols

3.当矩阵中没有值为 0 时，不可以调用 flip 函数

4.调用 flip 和 reset 函数的次数加起来不会超过 1000 次

示例 1：

输入: 
["Solution","flip","flip","flip","flip"]
[[2,3],[],[],[],[]]
输出: [null,[0,1],[1,2],[1,0],[1,1]]

示例 2：

输入: 
["Solution","flip","flip","reset","flip"]
[[1,2],[],[],[],[]]
输出: [null,[0,0],[0,1],null,[0,0]]

输入语法解释：

输入包含两个列表：被调用的子程序和他们的参数。Solution 的构造函数有两个参数，分别为 n_rows 和 n_cols。flip 和 reset 没有参数，参数总会以列表形式给出，哪怕该列表为空。

解题思路：

首先要充分了解本题的题意，一开始我也理解错了，以为是随机产生二维坐标，忽略了重复这个点。本题是经典的产生无重复数序列的随机数。假设矩阵的规模是n*m，对于每一个格子都对应着一个编号num=m*(i-1)+j，其中i是当前行，j是当前列。那么该题就转换成如何产生一个[0，n*m-1]的随机数，无重复。

一开始，想法比较简单，认为将已经产生的随机数放入哈希表，每一次都产生一个[0，n*m-1]的随机数，直到hash表中没有的值。很明显超时了，原因在于假如区间是[0,99]，已经产生了99个数，只剩下一个数，那么能够随机到剩下一个数的概率是0.01，一般而言，理论上需要100次，但是运气不好的话可能永远都不是！可以用根据概率论的方法，随机查询次数的期望是：n*m/1+n*m/2+n*m/3+n*m/4+n*m/5+......+n*m/n*m。算法时间复杂度期望的范围是o(n) --- O(n^2)。

很显然，我们不能靠运气使得复杂度变成o(n)，仔细分析造成算法变慢的原因在于产生了重复的数。使用映射的方法可以很好地避免重复。假如第i次随机，产生[0,n*m-i]之间的一个随机数r，如果r已经被使用，那么使用r的映射，如果r的映射也被使用，那么继续查找映射的映射。每次只需要将r与n*m-1建立映射关系即可，因为下一次必然不可能随机到n*m-1，这样一来就巧妙地避免了重复。

 

                                 

 

C++代码

class Solution {
public:
    Solution(int n_rows, int n_cols) {
        total = n_rows * n_cols;
        row = n_rows;
        col = n_cols;
    }
    vector<int> flip() {
        if (total <= 0) return{};
        int r = rand() % (total--);
        int x ;
        if (m.count(r) > 0) {
            x = m[r];
        }
        else x = r;
        m[r] = m.count(total) > 0 ? m[total] : total;
        return{ x / col,x%col };
    }
    void reset() {
        m.clear();
        total = row * col;
    }
private:
    unordered_map<int, int> m;
    int total;
    int row, col;
};

static const auto io_sync_off = []() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    return nullptr;
}();