384. 打乱数组
给你一个整数数组 nums ，设计算法来打乱一个没有重复元素的数组。打乱后，数组的所有排列应该是 等可能 的。
实现 Solution class:
Solution(int[] nums) 使用整数数组 nums 初始化对象
int[] reset() 重设数组到它的初始状态并返回
int[] shuffle() 返回数组随机打乱后的结果
示例 1：
1
2
3
4
5
6
7
8
9
10
| 输入
["Solution", "shuffle", "reset", "shuffle"]
[[[1, 2, 3]], [], [], []]
输出
[null, [3, 1, 2], [1, 2, 3], [1, 3, 2]]
解释
Solution solution = new Solution([1, 2, 3]);
solution.shuffle(); // 打乱数组 [1,2,3] 并返回结果。任何 [1,2,3]的排列返回的概率应该相同。例如，返回 [3, 1, 2]
solution.reset(); // 重设数组到它的初始状态 [1, 2, 3] 。返回 [1, 2, 3]
solution.shuffle(); // 随机返回数组 [1, 2, 3] 打乱后的结果。例如，返回 [1, 3, 2]
|
提示：
朴素思想
每次从剩余的原数组中拿一个出来到新数组，直到拿完为止
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
| import "math/rand"
type Solution struct {
nums, original []int
}
func Constructor(nums []int) Solution {
return Solution{nums, append([]int(nil), nums...)}
}
func (s *Solution) Reset() []int {
copy(s.nums, s.original)
return s.nums
}
func (s *Solution) Shuffle() []int {
shuffled := make([]int, len(s.nums))
for i := range shuffled {
j := rand.Intn(len(s.nums))
shuffled[i] = s.nums[j]
s.nums = append(s.nums[:j], s.nums[j+1:]...)
}
s.nums = shuffled
return s.nums
}
|
Fisher-Yates 洗牌算法
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
| type Solution struct {
nums, original []int
}
func Constructor(nums []int) Solution {
return Solution{nums, append([]int(nil), nums...)}
}
func (s *Solution) Reset() []int {
copy(s.nums, s.original)
return s.nums
}
func (s *Solution) Shuffle() []int {
n := len(s.nums)
for i := range s.nums {
j := i + rand.Intn(n-i)
s.nums[i], s.nums[j] = s.nums[j], s.nums[i]
}
return s.nums
}
|
