力扣 384. 打乱数组 随机洗牌算法

https://leetcode-cn.com/problems/shuffle-an-array/

思路：随机洗牌算法可以等概率的打乱输入，简单介绍一下经典洗牌算法$Knuth-Shuffle$。它的思路就是逐次遍历数组的每一位，对于第$i$位，随机选取$0<=j<=i$，并交换$a_i、a_j$。下面从数学角度上解释一下为什么这个算法是等概率的：对于第$i$个元素，其最终位置$j$要么$>=i$要么$<i$，那么我们分别考虑两种情况下的概率（假设下标从0开始）：

1. $i<=j$，那么一定是在第$j$次随机中被选中，剩余的$n-j-1$次随机中未被选中，概率$P=\frac{1}{j+1}\ast (1-\frac{1}{j+2})\ast ...\ast (1-\frac{1}{n})=\frac{1}{j+1}\ast \frac{j+1}{j+2}\ast ...\ast \frac{n-1}{n}=\frac{1}{n}$。
2. $i>j$，那么一定是在第$i$次随机中选中位置$j$，剩余的$n-i-1$次随机中未选中位置$j$，概率$P=\frac{1}{i+1}\ast (1-\frac{1}{i+2})\ast ...\ast (1-\frac{1}{n})=\frac{1}{i+1}\ast \frac{i+1}{i+2}\ast ...\ast \frac{n-1}{n}=\frac{1}{n}$。

综上，第$i$个元素分配到任意一个位置的概率都是相等的，因此该算法是等概率的。

class Solution {
public:
    Solution(vector<int>& nums) {
        this->nums=nums;
    }
    
    vector<int> reset() {
        return nums;
    }
    
    vector<int> shuffle() {
        vector<int> ans = nums;
        int len=ans.size();
        for(int i=1;i<len;i++)
        {
            int idx=rand()%(i+1);
            swap(ans[i], ans[idx]);
        }
        return ans;
    }
private:
    vector<int> nums;
};

/**
 * 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();
 */