LeetCode-Solutions/Python/shuffle-an-array.py at master · Monika-R/LeetCode-Solutions · GitHub

38 lines (26 loc) · 687 Bytes

# Time: O(n)

# Space: O(n)

import random

class Solution(object):

def __init__(self, nums):

"""

:type nums: List[int]

:type size: 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 = list(self.__nums)

for i in xrange(len(nums)):

j = random.randint(i, len(nums)-1)

nums[i], nums[j] = nums[j], nums[i]

return nums

Expand file tree