算法(TS):打乱数组

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

实现 Solution class:

  • Solution(int\[\] nums) 使用整数数组 nums 初始化对象
  • int\[\] reset() 重设数组到它的初始状态并返回
  • int\[\] shuffle() 返回数组随机打乱后的结果

解释

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 <= 50
  • -106 <= numsi <= 106
  • nums 中的所有元素都是 唯一的
  • 最多可以调用 104 次 reset 和 shuffle

解法一

用 Math.random() 生成随机数从 nums 取值,将取出来的值从数组中移除,添加到新数组中,当 nums 中的数取完之后,将新数组返回。

ts 复制代码
class Solution {
    private nums: number[] = []
    constructor(nums: number[]) {
        this.nums = nums
    }

    reset(): number[] {
        return this.nums
    }

    shuffle(): number[] {
        const thisNums = this.nums.concat()
        const result: number[] = []
        while(thisNums.length) {
            let i = Math.floor(Math.random() * thisNums.length)
            result.push(thisNums[i])
            thisNums.splice(i,1)
        }
        return result

    }
}

空间复杂度O(n),时间复杂度O(n * n)

解法二:洗牌算法

用 Math.random() 生成坐标从 nums 取值,将取出来的数与数组最后一个元素交换,接下来从剩下的数字中取值。

ts 复制代码
class Solution {
    private nums: number[] = []
    constructor(nums: number[]) {
        this.nums = nums
    }

    reset(): number[] {
        return this.nums
    }

    shuffle(): number[] {
        const shuffle = this.nums.concat()
        let lastIndex = result.length - 1
        while(lastIndex > 0) {
            const i = Math.floor(Math.random() * (lastIndex + 1))
            const temp = shuffle[i]
            shuffle[i] = shuffle[lastIndex]
            shuffle[lastIndex] = temp
            lastIndex--
        }
        return shuffle

    }
}

时间复杂度O(n),空间复杂度O(n)

相关推荐
aWty_4 小时前
实分析入门(12)--可测函数
学习·数学·算法·实变函数
海砥装备HardAus4 小时前
无人机姿态解算中「重力矢量观测退化」机理与动态补偿技术
算法·无人机·飞控
广州灵眸科技有限公司4 小时前
瑞芯微RV1126B开发板(EASY-EAI-PI2) 开发套件组装上电
网络·数据库·人工智能·算法·飞书
SuperHeroWu74 小时前
【算法】强化学习中奖励和损失函数的关系
算法·环境·强化学习·损失函数·奖励
voidmort4 小时前
9. 微调(Fine-tuning)的数学原理
人工智能·算法·机器学习
晚风吹红霞5 小时前
C++ stack 和 queue 完全指南:适配器模式与双端队列的奥秘
c++·算法·适配器模式
casual~5 小时前
十六届蓝桥杯国赛个人题解
经验分享·学习·算法·蓝桥杯
方也_arkling5 小时前
【Java-Day18】API篇-Arrays
java·算法·排序算法
吴可可1235 小时前
Curve.GetSplitCurves高效分割技巧
算法