1. 題目

給定一個可能含有重復元素的整數數組，要求隨機輸出給定的數字的索引。 您可以假設給定的數字一定存在于數組中。

注意：
數組大小可能非常大。 使用太多額外空間的解決方案將不會通過測試。

示例: 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);

來源：力扣（LeetCode） 鏈接：https://leetcode-cn.com/problems/random-pick-index
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 蓄水池抽樣

類似題目：LeetCode 519. 隨機翻轉矩陣（哈希）

• 遇到 target，計數+1，然后用 計數分之1的概率去替換之前找到的 target 的下標

class Solution {vector<int> arr;int i, idx, count;bool Occur; public:Solution(vector<int>& nums) {arr = nums;}int pick(int target) {Occur = false;count = 0;for(i = 0; i < arr.size(); ++i){if(arr[i] == target){if(!Occur){idx = i;Occur = true;count = 1;}else{count++;if(rand()%count == 0)//有 1/count 的概率等于0idx = i;}}}return idx;} };

總結

以上是生活随笔為你收集整理的LeetCode 398. 随机数索引（概率）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。