「 每日一练，快乐水题 」398. 随机数索引

Posted 2022-06-07 谁吃薄荷糖

文章目录

• • 🔴力扣原题：
  • 🟠题目简述：
  • 🟡解题思路：
  • 🟢C++代码：
  • 🔵结果展示：

---

🔴力扣原题：

398. 随机数索引

🟠题目简述：

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

实现 Solution 类：

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

🟡解题思路：

1. 哈希大法好;
2. 利用哈希map把重复数的索引记录；
3. 根据target对应索引数构造随机数；
4. 返回随机索引；
5. 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);
 */

🔵结果展示：