leetcode 384. 打乱数组 medium

leetcode 384. 打乱数组  medium          

题目描述：

打乱一个没有重复元素的数组。

示例:

// 以数字集合 1, 2 和 3 初始化数组。
int[] nums = {1,2,3};
Solution solution = new Solution(nums);

// 打乱数组 [1,2,3] 并返回结果。任何 [1,2,3]的排列返回的概率应该相同。
solution.shuffle();

// 重设数组到它的初始状态[1,2,3]。
solution.reset();

// 随机返回数组[1,2,3]打乱后的结果。
solution.shuffle();

解题思路：

洗牌算法，怎么洗一堆牌？先从中抽一张，扔到最后去，再从剩下的抽一张，扔到倒数第二位置去。

代码：

//
class Solution {
public:
    Solution(vector& nums):vec(nums) {
        
    }
    
    /** Resets the array to its original configuration and return it. */
    vector reset() {
        return vec;
    }
    
    /** Returns a random shuffling of the array. */
    vector shuffle() {
        vector res(vec);
        for(int i=res.size()-1;i>=0;--i){
            swap(res[i],res[rand() % (i+1)]);
        }
        
        return res;
    }
    
    vector vec;
    
};

/**
 * Your Solution object will be instantiated and called as such:
 * Solution* obj = new Solution(nums);
 * vector param_1 = obj->reset();
 * vector param_2 = obj->shuffle();
 */