(leetcode算法题)384. 打乱数组 398. 随机数索引

问题转化:

题目要求将nums中的数字出现的次序随机打乱

转化成:对于 0 号位置来说,nums[i], ..., nums[n - 1] 可以等概率的出现 ... && ...

&& 对于 n - 1号位置来说,nums[i], ..., nums[n - 1] 可以等概率的出现

转化成:每个位置从nums[i], ..., nums[n - 1]中不放回重复抽样,

对于每个位置来说,随机抽取一个数字之后,后面的抽取不再考虑这个数字

那么这个题就是不放回抽样的的典型例子,

需要注意的就是没有当前被抽到的数字要全部放到数组的后面

代码如下

cpp 复制代码
class Solution {
private:
    vector<int> _nums;
public:
    Solution(vector<int>& nums) {
        // srand(nullptr);
        _nums = nums;
    }
    
    vector<int> reset() {
        return _nums;
    }
    
    vector<int> shuffle() {
        vector<int> snums(_nums);
        int n = snums.size();
        for(int i = 0; i < n; ++i){
// 抽样不放回,每次则每个数放到i位置的概率为 1 / n
            swap(snums[i], snums[i + rand() % (n - i)]);
        }
        return snums;
    }
};

/**
 * 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();
 */

这个题的测试用例对损耗内存的要求不是非常严格,但是对效率的要求很高

应该使用hash表把数字和数字对应的下标管理起来

相关推荐
Wendy14417 小时前
【线性回归(最小二乘法MSE)】——机器学习
算法·机器学习·线性回归
拾光拾趣录8 小时前
括号生成算法
前端·算法
渣呵8 小时前
求不重叠区间总和最大值
算法
拾光拾趣录9 小时前
链表合并:双指针与递归
前端·javascript·算法
好易学·数据结构9 小时前
可视化图解算法56:岛屿数量
数据结构·算法·leetcode·力扣·回溯·牛客网
香蕉可乐荷包蛋10 小时前
AI算法之图像识别与分类
人工智能·学习·算法
chuxinweihui10 小时前
stack,queue,priority_queue的模拟实现及常用接口
算法
tomato0910 小时前
河南萌新联赛2025第(一)场:河南工业大学(补题)
c++·算法
墨染点香10 小时前
LeetCode Hot100【5. 最长回文子串】
算法·leetcode·职场和发展
甄卷12 小时前
李沐动手学深度学习Pytorch-v2笔记【08线性回归+基础优化算法】2
pytorch·深度学习·算法