LeetCode-384 打乱数组

import random
# 打乱一个没有重复元素的数组。
#
# 示例:
# // 以数字集合 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();


class Solution:

    def __init__(self, nums):
        self.nums = nums
        self.save_nums = nums

    def reset(self):
        """
        Resets the array to its original configuration and return it.
        """
        return self.save_nums

    def shuffle(self):
        """
        Returns a random shuffling of the array.
        """
        lens = len(self.nums)
        array = []
        while len(array) != lens:
            key = random.randint(0, lens-1)
            # rondom.randint: [a,b], numpy.random.randint: [a, b)
            if self.nums[key] not in array:
                array.append(self.nums[key])
        return array