398. Random Pick Index

最新推荐文章于 2024-10-06 07:15:00 发布

原创最新推荐文章于 2024-10-06 07:15:00 发布 · 233 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

429 篇文章

订阅专栏

本文介绍了一种简单随机采样算法的实现方法，通过遍历数组并收集目标元素的所有出现位置，最后从中随机选择一个返回。该算法适用于内存限制的情况，避免了使用额外的数据结构来存储位置信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

简单，一开始打算用map存储位置，发现空间不足，因此没有存储了。采用了直接暴力求解的方法。

class Solution {
private:
    vector<int> nums;
    //map<int,vector<int>> index;
public:
    Solution(vector<int> nums) {
        this->nums=nums;
        /*for(int i=0;i<nums.size();i++)
            index[nums[i]].push_back(i);*/
    }

    int pick(int target) {

        vector<int> targetIndex;

        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]==target)
                targetIndex.push_back(i);
        }

        int index=rand()%targetIndex.size();

        return targetIndex[index];

    }
};

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