[LeetCode] 384. Shuffle an Array 數組洗牌
阿新 • • 發佈:2018-10-03
個數字 splay random 復數 clone The cat leet new
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();
給一個沒有重復數字的數組,實現重置和洗牌的功能。
解法:遍歷數組每個位置,每次都隨機生成一個坐標位置,然後交換當前位置和隨機位置的數字,這樣如果數組有n個數字,那麽也隨機交換了n組位置,從而達到了洗牌的目的。
Java:
import java.util.Random; public class Solution { private int[] nums; private Random random; public Solution(int[] nums) { this.nums = nums; random = new Random(); } /** Resets the array to its original configuration and return it. */ public int[] reset() { return nums; } /** Returns a random shuffling of the array. */ public int[] shuffle() { if(nums == null) return null; int[] a = nums.clone(); for(int j = 1; j < a.length; j++) { int i = random.nextInt(j + 1); swap(a, i, j); } return a; } private void swap(int[] a, int i, int j) { int t = a[i]; a[i] = a[j]; a[j] = t; } }
Python:
import random class Solution(object): def __init__(self, nums): """ :type nums: List[int] :type size: int """ self.__nums = nums def reset(self): """ Resets the array to its original configuration and return it. :rtype: List[int] """ return self.__nums def shuffle(self): """ Returns a random shuffling of the array. :rtype: List[int] """ nums = list(self.__nums) for i in xrange(len(nums)): j = random.randint(i, len(nums)-1) nums[i], nums[j] = nums[j], nums[i] return nums # Your Solution object will be instantiated and called as such: # obj = Solution(nums) # param_1 = obj.reset() # param_2 = obj.shuffle()
C++:
class Solution { public: Solution(vector<int> nums): v(nums) {} /** Resets the array to its original configuration and return it. */ vector<int> reset() { return v; } /** Returns a random shuffling of the array. */ vector<int> shuffle() { vector<int> res = v; for (int i = 0; i < res.size(); ++i) { int t = i + rand() % (res.size() - i); swap(res[i], res[t]); } return res; } private: vector<int> v; };
[LeetCode] 384. Shuffle an Array 數組洗牌