跳到主要内容

384. 打乱数组

给你一个整数数组 nums ，设计算法来打乱一个没有重复元素的数组。打乱后，数组的所有排列应该是 等可能 的。

实现 Solution class:

  • Solution(int[] nums) 使用整数数组 nums 初始化对象

  • int[] reset() 重设数组到它的初始状态并返回

  • int[] shuffle() 返回数组随机打乱后的结果

示例 1：

输入
["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 <= nums.length <= 200

  • -106 <= nums[i] <= 106

  • nums 中的所有元素都是 唯一的

  • 最多可以调用 5 * 104 次 reset 和 shuffle

朴素思想​

每次从剩余的原数组中拿一个出来到新数组，直到拿完为止

image-20220308224732103

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
}

image-20220308225113061

Fisher-Yates 洗牌算法​

image-20220308233505113

image-20220308233512146

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
}

image-20220308233519522