C++刷leetcode（398. 随机数索引）【抽样】

糖逗

题目描述：

1. 给定一个可能含有重复元素的整数数组，要求随机输出给定的数字的索引。 您可以假设给定的数字一定存在于数组中。
   
2. 
   
3. 注意：
   
4. 数组大小可能非常大。 使用太多额外空间的解决方案将不会通过测试。
   
5. 
   
6. 示例:
   
7. 
   
8. int[] nums = new int[] {1,2,3,3,3};
   
9. Solution solution = new Solution(nums);
   
10. 
    
11. // pick(3) 应该返回索引 2,3 或者 4。每个索引的返回概率应该相等。
    
12. solution.pick(3);
    
13. 
    
14. // pick(1) 应该返回 0。因为只有nums[0]等于1。
    
15. solution.pick(1);

复制代码

1. class Solution {
   
2. public:
   
3.     vector<int> store;
   
4.     Solution(vector<int>& nums) {
   
5.         store = nums;
   
6.     }
   
7.    
   
8.     int pick(int target) {
   
9.         vector<int> temp;
   
10.         for(int i = 0; i < store.size(); i++){
    
11.             if(store[i] == target){
    
12.                 temp.push_back(i);
    
13.             }
    
14.         }
    
15.         int size = temp.size();
    
16.         int index = rand() % size;
    
17.         return temp[index];
    
18.     }
    
19. };
    

复制代码

注意事项：
1.rand（）函数的参考链接：https://blog.csdn.net/SHAOYEZUIZUISHAUI/article/details/100101114