leetcode 384 打乱数组

https://leetcode-cn.com/problems/shuffle-an-array/
给你一个整数数组nums，设计算法来打乱一个没有重复元素的数组。

实现Solutionclass：

• 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

这题的话其实就是考怎么实现STL里的random_shuffle算法。这里我们先考虑解决两个基础的问题，即怎么实现这个reset来返回初始状态。而由于随机交换本身是不可逆的，因此我们的Solution类中实际上要有两个vector，一个用来存储最开始初始化后的数组，在初始化后保持不变，以供在调用reset方法时返回数组的初始状态；那么另一个数组就可以用来存储经排列打乱后的状态。

vector<int> nums;
vector<int> original;

Solution(vector<int>& nums) {
    this->nums = nums;
    this->original = nums;
}

vector<int> reset() {
    return original;
}

现在来解决怎么随机排列的问题。考虑一种最直观的做法，假如一个数组有n个数，随机排列无非就是每次不放回地从里面拿出一个，拿n次而已。据此我们可以写出如下的算法：

vector<int> shuffle() {
    vector<int> shuffled = vector<int>(nums.size());
    vector<int> lst(nums.begin(), nums.end());
  
    for (int i = 0; i < nums.size(); ++i) {
        int j = rand() % (lst.size());
        auto it = lst.begin();
        advance(it, j);
        shuffled[i] = *it;
        lst.erase(it);
    }
    return shuffled;
}

注意这里由于涉及到一个不放回抽样的问题，因此需要用到迭代器。

思路二

将上述过程改进，做成Fisher-Yates洗牌算法。可以发现在上一步排序过程中我们用到了两个临时数组，一个用来抽，一个用来放，但这两个其实都是不需要的，我们可以直接来进行原地排序。

这个"原地"的思想有点类似于堆排序，具体来说，我们把每次随机选择到的元素放在数组最后面(或者最前面)即可。相较来说堆排序是把最大或者最小的元素放在最后面。

vector<int> shuffle() {
    for (int i = 0; i < nums.size(); i++) {
        int j = i + rand() % (nums.size() - i);
        swap(nums[i], nums[j]);
    }
    return nums;
}