前言

Hello！小伙伴！

非常感谢您阅读海轰的文章，倘若文中有错误的地方，欢迎您指出～

自我介绍 ଘ(੭ˊᵕˋ)੭

昵称：海轰

标签：程序猿｜C++选手｜学生

简介：因C语言结识编程，随后转入计算机专业，获得过国家奖学金，有幸在竞赛中拿过一些国奖、省奖...已保研。

学习经验：扎实基础 + 多做笔记 + 多敲代码 + 多思考 + 学好英语！唯有努力💪

文章仅作为自己的学习笔记用于知识体系建立以及复习

知其然知其所以然！

每日一题

398. 随机数索引

给定一个可能含有重复元素的整数数组，要求随机输出给定的数字的索引。您可以假设给定的数字一定存在于数组中。

注意：数组大小可能非常大。使用太多额外空间的解决方案将不会通过测试。

示例:


int[] nums = new int[] {1,2,3,3,3};
Solution solution = new Solution(nums);

// pick(3) 应该返回索引 2,3 或者 4。每个索引的返回概率应该相等。
solution.pick(3);

// pick(1) 应该返回 0。因为只有nums[0]等于1。
solution.pick(1);

思路：

使用哈希表存储每一个值对应的索引
查找时在值对应的索引数组中随机取一个元素

class Solution {
    unordered_map<int, vector<int>> pos;
public:
    Solution(vector<int> &nums) {
        for (int i = 0; i < nums.size(); ++i) {
            pos[nums[i]].push_back(i);
        }
    }

    int pick(int target) {
        auto &indices = pos[target];
        return indices[rand() % indices.size()];
    }
};

思路：（水塘抽样来源官方题解）

class Solution {
private:
    vector<int> nums;
public:
    Solution(vector<int>& nums) : nums(nums){

    }
    
    int pick(int target) {
        int ans = 0;
        for(int i = 0, cnt = 0; i < nums.size(); ++i) {
            if(nums[i] == target) {
                ++cnt;
                if(rand() % cnt == 0) {
                    ans = i;
                }
            }
        }
        return ans;
    }
};

类似使用水塘抽样题目：#### 382. 链表随机节点

结语

文章仅作为个人学习笔记记录，记录从0到1的一个过程

希望对您有一点点帮助，如有错误欢迎小伙伴指正

在这里插入图片描述

日新计划 · 04-11

前言

每日一题

398. 随机数索引

结语