(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表把数字和数字对应的下标管理起来

相关推荐
I'm a winner1 小时前
基于YOLO算法的医疗应用专题:第一章 计算机视觉与深度学习概述
算法·yolo·计算机视觉
vir022 小时前
P1928 外星密码(dfs)
java·数据结构·算法·深度优先·1024程序员节
喜欢吃燃面2 小时前
数据结构算法题:list
开发语言·c++·学习·算法·1024程序员节
寂静山林2 小时前
UVa 12991 Game Rooms
算法·1024程序员节
Dream it possible!3 小时前
LeetCode 面试经典 150_链表_合并两个有序链表(58_21_C++_简单)
leetcode·链表·面试·1024程序员节
余俊晖3 小时前
RLVR训练多模态文档解析模型-olmOCR 2技术方案(模型、数据和代码均开源)
人工智能·算法·ocr·grpo
凉虾皮3 小时前
2024包河初中组
学习·算法·1024程序员节
m0_748233644 小时前
C++ 模板初阶:从函数重载到泛型编程的优雅过渡
java·c++·算法·1024程序员节
以己之4 小时前
11.盛最多水的容器
java·算法·双指针·1024程序员节
初级炼丹师(爱说实话版)5 小时前
算法面经常考题整理(3)大模型
算法