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_colsflip 和 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;
}();

猜你喜欢

转载自blog.csdn.net/qq_23523409/article/details/85077587