384. 打乱数组
打乱一个没有重复元素的数组。
示例:
// 以数字集合 1, 2 和 3 初始化数组。
int[] nums = {1,2,3};
Solution solution = new Solution(nums);
// 打乱数组 [1,2,3] 并返回结果。任何 [1,2,3]的排列返回的概率应该相同。
solution.shuffle();
// 重设数组到它的初始状态[1,2,3]。
solution.reset();
// 随机返回数组[1,2,3]打乱后的结果。
solution.shuffle();
解题思路:Fisher–Yates shuffle 算法。算法的原理是每次从当前的数组中随机取出一个元素添加到待生成数组的末尾,然后在原数组中删除掉该元素。重复上述过程即可生成一个随机的打乱数组。在代码实现上进行了优化,时间复杂度为O(n)。
Python3代码如下:
class Solution(object):
def __init__(self, nums):
"""
:type nums: List[int]
"""
self.nums = nums
def reset(self):
"""
Resets the array to its original configuration and return it.
:rtype: List[int]
"""
return self.nums
def shuffle(self):
"""
Returns a random shuffling of the array.
:rtype: List[int]
"""
nums_ = self.nums[:]
length = len(nums_)
for i in range(length):
n = random.randrange(i,length)
nums_[i],nums_[n] = nums_[n],nums_[i]
return nums_
# Your Solution object will be instantiated and called as such:
# obj = Solution(nums)
# param_1 = obj.reset()
# param_2 = obj.shuffle()