398. 随机数索引

2021-12-14 20:03:15 阅读：119 来源： 互联网

标签：index target nums int Random 数组索引随机数 398

题目

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/random-pick-index
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
在这里插入图片描述

代码

class Solution {
    int[] res;
    int[] nums;
    public Solution(int[] nums) {
       this.nums=nums;
    }

    public int pick(int target) {
        Random r = new Random();
        int n=0,index=0;

        // 前 k 个元素先默认选上
        for (int i = 0; i <nums.length ; i++) {
            if (nums[i]==target) {//每找到一个目标
                n++;//我们以1/n的概率留下该数据
                if (r.nextInt(n)==0) 
                    index=i;
            }
        }

        return index;
    }
}

要点

室友水池抽样法，对于每个num[i]=target，都有1/n的概率留下该索引
蓄水池抽样（Random Reservior Sampling) 保存整个nums数组。O(N)额外空间

标签：index,target,nums,int,Random,数组,索引,随机数,398
来源： https://blog.csdn.net/weixin_43871309/article/details/121938326

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

398. 随机数索引

题目

代码

要点