-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path384.py
38 lines (26 loc) · 906 Bytes
/
384.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
# Problem link: https://leetcode.com/problems/shuffle-an-array/
from random import choice
class Solution:
def __init__(self, nums: List[int]):
self.original = nums
def reset(self) -> List[int]:
"""
Resets the array to its original configuration and return it.
"""
return self.original
def shuffle(self) -> List[int]:
"""
Returns a random shuffling of the array.
"""
n = len(self.original)
indexes = list(range(0, n))
shuffled = []
for i in range(n):
chosenPos = choice(indexes)
shuffled.append(self.original[chosenPos])
indexes.remove(chosenPos)
return shuffled
# Your Solution object will be instantiated and called as such:
# obj = Solution(nums)
# param_1 = obj.reset()
# param_2 = obj.shuffle()