1.题目描述
给你一个整数数组 nums ,设计算法来打乱一个没有重复元素的数组。实现 Solution class:
Solution(int[] nums) 使用整数数组 nums 初始化对象
int[] reset() 重设数组到它的初始状态并返回
int[] shuffle() 返回数组随机打乱后的结果
示例:
提示:
(1)1 <= nums.length <= 200
(2)-10^6 <= nums[i] <= 10^6
(3)nums 中的所有元素都是 唯一的
(3)最多可以调用 5 * 104 次 reset 和 shuffle
2.思路( Fisher-Yates 洗牌算法)
在每次迭代中,生成一个范围在当前下标到数组末尾元素下标之间的随机整数。接下来,将当前元素和随机选出的下标所指的元素互相交换 - 这一步模拟了每次从 “帽子” 里面摸一个元素的过程,其中选取下标范围的依据在于每个被摸出的元素都不可能再被摸出来了。此外还有一个需要注意的细节,当前元素是可以和它本身互相交换的 - 否则生成最后的排列组合的概率就不对了。
3.代码
class Solution {
public:
Solution(vector<int>& nums):data(nums.begin(), nums.end()) {
}
/** Resets the array to its original configuration and return it. */
vector<int> reset() {
return data;
}
/** Returns a random shuffling of the array. */
vector<int> shuffle() {
vector<int> shuffled(data.begin(), data.end());
for(int i = 0; i < shuffled.size(); ++i){
swap(shuffled[i], shuffled[rand() % (i + 1)]);
}
return shuffled;
}
private:
vector<int> data;
};
/**
* 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();
*/
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(n)