Shuffle a set of numbers without duplicates.
Example:
// Init an array with set 1, 2, and 3. int[] nums = {1,2,3}; Solution solution = new Solution(nums); // Shuffle the array [1,2,3] and return its result. Any permutation of [1,2,3] must equally likely to be returned. solution.shuffle(); // Resets the array back to its original configuration [1,2,3]. solution.reset(); // Returns the random shuffling of array [1,2,3]. solution.shuffle();
===================================================================
题目链接:https://leetcode.com/problems/shuffle-an-array/
题目大意:洗牌和复原操作,要求洗牌得到结果等概率出现。
思路:主要是洗牌操作,随机生成一个位置跟最后一位交换,再随机生成一个位置跟倒数第二位交换,交换操作保证集合的正确性。
参考代码:
class Solution {
public:
Solution(vector<int> nums) {
this -> nums = nums ;
srand ( time ( 0 ) ) ;
}
/** Resets the array to its original configuration and return it. */
vector<int> reset() {
return nums ;
}
/** Returns a random shuffling of the array. */
vector<int> shuffle() {
vector <int> ans = nums ;
for ( int i = ans.size() - 1 ; i > 0 ; i -- )
swap ( ans[i] , ans[rand() % ( i + 1 )] ) ;
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();
*/