算法(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)

相关推荐
xieliyu.19 小时前
Java算法精讲:双指针(三)
java·开发语言·算法
一条小锦吕*19 小时前
基于Spring Boot + 数据可视化 + 协同过滤算法的推荐系统设计与实现(源码+论文+部署全讲解)
spring boot·算法·信息可视化
夜雪闻竹20 小时前
测试策略:单元测试 + 集成测试怎么写
typescript·单元测试·集成测试·chatcrystal
如竟没有火炬21 小时前
最大矩阵——单调栈
数据结构·python·线性代数·算法·leetcode·矩阵
8Qi821 小时前
LeetCode 1143 & 718:最长公共子序列 / 最长重复子数组
算法·leetcode·职场和发展·动态规划
绿算技术1 天前
万卡推理集群存储选型分析:从核心架构到应用视角
大数据·科技·算法·架构
想吃火锅10051 天前
【leetcode】1.两数之和js版
javascript·算法·leetcode
net3m331 天前
一阶软件低通滤波器算法
人工智能·算法
水木流年追梦1 天前
大模型入门-大模型优化方法12-YaRN 长文本外推技术
人工智能·分布式·算法·正则表达式·prompt