数据结构与算法之深入解析“随机翻转矩阵”的求解思路与算法示例

Posted 2022-03-23 Serendipity·y

一、题目要求

• 给你一个 m x n 的二元矩阵 matrix ，且所有值被初始化为 0，请你设计一个算法，随机选取一个满足 matrix[i][j] == 0 的下标 (i, j)，并将它的值变为 1，所有满足 matrix[i][j] == 0 的下标 (i, j) 被选取的概率应当均等。
• 尽量最少调用内置的随机函数，并且优化时间和空间复杂度。
• 实现 Solution 类：
• • Solution(int m, int n) 使用二元矩阵的大小 m 和 n 初始化该对象；
• • int[] flip() 返回一个满足 matrix[i][j] == 0 的随机下标 [i, j] ，并将其对应格子中的值变为 1；
• • void reset() 将矩阵中所有的值重置为 0。
• 示例：

输入
["Solution", "flip", "flip", "flip"