重点在于如何打乱数组
创建一个列表复制原数组,每次随机从列表中取出一个数字组成新的数组。这样做能实现要求,但比较耗时。
Fisher-Yates 洗牌算法
设待原地乱序的数组 nums。
循环 n 次,在第 i 次循环中(0 <= i <= n):
- 在 [i,n)中随机抽取一个下标 j;
- 将第 i 个元素与第 j个元素交换。
自己的(84ms)
class Solution {
int[] old;
public Solution(int[] nums) {
old = (int[])Arrays.copyOf(nums,nums.length);
}
public int[] reset() {
return old;
}
public int[] shuffle() {
int[] n = new int[old.length];
List<Integer> l = Arrays.stream(old).boxed().collect(Collectors.toList());
Random r = new Random();
for(int i = 0; i < old.length; i++){
int j = r.nextInt(l.size());
n[i] = l.get(j);
l.remove(j);
}
return n;
}
}
题解(68ms)
class Solution {
int[] nums;
int[] original;
public Solution(int[] nums) {
this.nums = nums;
this.original = new int[nums.length];
System.arraycopy(nums, 0, original, 0, nums.length);
}
public int[] reset() {
System.arraycopy(original, 0, nums, 0, nums.length);
return nums;
}
public int[] shuffle() {
Random random = new Random();
for (int i = 0; i < nums.length; ++i) {
int j = i + random.nextInt(nums.length - i);
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
return nums;
}
}