问题转化:
题目要求将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 / nswap(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表把数字和数字对应的下标管理起来