问题转化:
题目要求将nums中的数字出现的次序随机打乱
转化成:对于 0 号位置来说,nums[i], ..., nums[n - 1] 可以等概率的出现 ... && ...
&& 对于 n - 1号位置来说,nums[i], ..., nums[n - 1] 可以等概率的出现
转化成:每个位置从nums[i], ..., nums[n - 1]中不放回重复抽样,
对于每个位置来说,随机抽取一个数字之后,后面的抽取不再考虑这个数字
那么这个题就是不放回抽样的的典型例子,
需要注意的就是没有当前被抽到的数字要全部放到数组的后面
代码如下
class Solution {
private:
vector<int> _nums;
public:
Solution(vector<int>& nums) {
// srand(nullptr);
_nums = nums;
}
vector<int> reset() {
return _nums;
}
vector<int> shuffle() {
vector<int> snums(_nums);
int n = snums.size();
for(int i = 0; i < n; ++i){
// 抽样不放回,每次则每个数放到i位置的概率为 1 / n
swap(snums[i], snums[i + rand() % (n - i)]);
}
return snums;
}
};
/**
* Your Solution object will be instantiated and called as such:
* Solution* obj = new Solution(nums);
* vector<int> param_1 = obj->reset();
* vector<int> param_2 = obj->shuffle();
*/
这个题的测试用例对损耗内存的要求不是非常严格,但是对效率的要求很高
应该使用hash表把数字和数字对应的下标管理起来