题目:
给定一个可能含有重复元素的整数数组,要求随机输出给定的数字的索引。
您可以假设给定的数字一定存在于数组中。
注意:
数组大小可能非常大。 使用太多额外空间的解决方案将不会通过测试。
示例:
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);
思路:
生成随机数,
等于1/i的概率时,就替换结果
解答:
/**
* @param {number[]} nums
*/
var Solution = function(nums) {
// 构造函数记录下数组
this.nums = nums;
};
/**
* @param {number} target
* @return {number}
*/
Solution.prototype.pick = function(target) {
// 注意初始n 为1
let n = 1;
let res = 0;
for(let i = 0;i < this.nums.length;i++){
let num = this.nums[i];
if(num === target){
// [0,1)
let r = Math.random(1);
let p = Math.floor(r * n);
// 如果有1/n的概率被选中
if(p === 0){
res = i;
}
n++;
}
}
return res;
};
/**
* Your Solution object will be instantiated and called as such:
* var obj = new Solution(nums)
* var param_1 = obj.pick(target)
*/