[数据结构与算法] 「每日一练，快乐水题」398. 随机数索引

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 数据结构与算法 -> 「每日一练，快乐水题」398. 随机数索引 -> 正文阅读

[数据结构与算法]「每日一练，快乐水题」398. 随机数索引

文章目录

🔴力扣原题：

398. 随机数索引

🟠题目简述：

给你一个可能含有重复元素的整数数组 nums ，请你随机输出给定的目标数字 target 的索引。你可以假设给定的数字一定存在于数组中。

实现 Solution 类：

Solution(int[] nums) 用数组 nums 初始化对象。
int pick(int target) 从 nums 中选出一个满足 nums[i] == target 的随机索引 i 。如果存在多个有效的索引，则每个索引的返回概率应当相等。

🟡解题思路：

哈希大法好;
利用哈希map把重复数的索引记录；
根据target对应索引数构造随机数；
返回随机索引；
over;

🟢C++代码：

class Solution {
private:
    unordered_map<int, vector<int>> umap;
public:
    Solution(vector<int>& nums) {
        for(int i = 0; i < nums.size(); i++)
        {
            umap[nums[i]].push_back(i);
        }
    }
    
    int pick(int target) {
        auto & vecNums = umap[target];
        int randNum = rand() % vecNums.size();
        return vecNums[randNum];
    }
};

/**
 * Your Solution object will be instantiated and called as such:
 * Solution* obj = new Solution(nums);
 * int param_1 = obj->pick(target);
 */