本文主要是介绍LeetCode 519. 随机翻转矩阵,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目链接:
力扣https://leetcode-cn.com/problems/random-flip-matrix/
【超时】首先想到的是把二位数组一纬话,因为行和列的范围都是10^4,所以行*10^4+列即可。然后把这些坐标存在一个一纬数组中,每次随机选取选的是数组的下标,然后将这个下标下的数和后面没有被选到的数交换,并记录最后一个没被选择过的数的下标end。再次选取时从0到这个下标end中选即可。但是这样会超时,主要原因是初始化时需要行*列个数,即10^8。下面是超时的代码:
class Solution:def __init__(self, m: int, n: int):self.arr = []self.m, self.n = m, nfor i in range(m):for j in range(n):self.arr.append(i * 10000 + j)self.end = m * n - 1def flip(self) :import randomindex = random.randint(0, self.end)[x, y] = self.arr[index] / 10000, self.arr[index] % 10000self.arr[index], self.arr[self.end] = self.arr[self.end], self.arr[index]self.end -= 1return [int(x), y]def reset(self) -> None:self.end = self.m * self.n - 1
【优化】那么这个一纬数组如何优化呢,显然可以用字典来保存,因为最初是初始化为和自身有关的数,所以字典在初始化时可以不填值,把填初始值的操作延后在每个数的第一次随机取数中。
class Solution:def __init__(self, m: int, n: int):self.arr = {}self.m, self.n = m, nself.end = m * n - 1def flip(self) :import randomindex = random.randint(0, self.end)select = self.arr.get(index, index)[x, y] = select / self.n, select % self.nself.arr[index] = self.arr.get(self.end, self.end) #这里特别要注意!#因为self.end前面可能被抽到过!所以这里要把index中更新为self.end修改后的值!!self.end -= 1return [int(x), y]def reset(self) -> None:self.arr = {}self.end = self.m * self.n - 1
这其实是昨天的每日一题,不是昨天不想写,实在是给干懵了
这篇关于LeetCode 519. 随机翻转矩阵的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!