leetcode 384：打乱数组

leetcode 384：打乱数组

给你一个整数数组 nums ，设计算法来打乱一个没有重复元素的数组。打乱后，数组的所有排列应该是 等可能 的。

实现 Solution class:

• Solution(int[] nums) 使用整数数组 nums 初始化对象

• int[] reset() 重设数组到它的初始状态并返回

• int[] shuffle() 返回数组随机打乱后的结果

示例 1：

输入
["Solution", "shuffle", "reset", "shuffle"]
[[[1, 2, 3]], [], [], []]
输出
[null, [3, 1, 2], [1, 2, 3], [1, 3, 2]]
​
解释
Solution solution = new Solution([1, 2, 3]);
solution.shuffle();    // 打乱数组 [1,2,3] 并返回结果。任何 [1,2,3]的排列返回的概率应该相同。例如，返回 [3, 1, 2]
solution.reset();      // 重设数组到它的初始状态 [1, 2, 3] 。返回 [1, 2, 3]
solution.shuffle();    // 随机返回数组 [1, 2, 3] 打乱后的结果。例如，返回 [1, 3, 2]

提示：

• 1 <= nums.length <= 200

• -106 <= nums[i] <= 106

• nums 中的所有元素都是 唯一的

• 最多可以调用 5 * 104 次 reset 和 shuffle

Related Topics

数组

数学

随机化

思路1：

将数组中的元素放入list中，使用random随机下标抽取。时间复杂度为O(n^2)

class Solution {
    private int[] nums;
    public Solution(int[] nums) {
        this.nums = new int[nums.length];
        for(int i = 0 ; i < nums.length;i++){
            this.nums[i] = nums[i];
        }
    }
    
    public int[] reset() {
        int[] result = new int[nums.length];
        for(int i = 0 ; i <nums.length;i++){
            result[i] = nums[i];
        }
        return result;
    }
    
    public int[] shuffle() {
        ArrayList<Integer> list = new ArrayList<>();
        for(int i = 0 ; i < nums.length;i++){
            list.add(nums[i]);
        }
        int[] result = new int[nums.length];
        Random random = new Random();
        for(int i = 0 ; i < nums.length;i++){
            int ran = random.nextInt(list.size());
            result[i] = list.remove(ran);
        }
        return result;
    }
}
解答成功:
            执行耗时:88 ms,击败了29.23% 的Java用户
            内存消耗:49.5 MB,击败了5.32% 的Java用户

思路2：参考leetcode官方

• 在 [i,n) 中随机抽取一个下标 j

• 将第 i个元素与第 j个元素交换。

• 可以达到O(N)的时间复杂度。

  class Solution {
      private int[] nums;
      public Solution(int[] nums) {
          this.nums = new int[nums.length];
          for(int i = 0 ; i < nums.length;i++){
              this.nums[i] = nums[i];
          }
      }
      
      public int[] reset() {
          int[] result = new int[nums.length];
          for(int i = 0 ; i <nums.length;i++){
              result[i] = nums[i];
          }
          return result;
      }
      
      public int[] shuffle() {
          int[] result = new int[nums.length];
          for(int i = 0 ; i < nums.length;i++){
              result[i] = nums[i];
          }
          Random random = new Random();
          for(int i = 0 ; i < nums.length;i++){
              //生成一个[i,n)的随机数
              int index = i + random.nextInt(nums.length-i);
              //交换i和index位置的值 此时index就是随机选中的数字
              int temp = result[index];
              result[index] = result[i];
              result[i] = temp;
          }
          return result;
      }
  }
      解答成功:
          执行耗时:72 ms,击败了99.71% 的Java用户
          内存消耗:49.4 MB,击败了5.32% 的Java用户