题目描述
打乱一个没有重复元素的数组。
示例:
// 以数字集合 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:
vector<int> stable;
Solution(vector<int>& nums) {
int len = nums.size();
for(int i=0;i<len;i++) stable.push_back(nums[i]);
}
/** Resets the array to its original configuration and return it. */
vector<int> reset() {
return stable;
}
/** Returns a random shuffling of the array. */
vector<int> shuffle() {
vector<int> shu = stable;
int len = shu.size();
for(int i=0;i<len;++i) swap(shu[rand()%len],shu[i]);
return shu;
}
};