数据结构
网络
RDBMS
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 语言编程
C++
C#
MongoDB
MySQL
JavaScript
PHP
物理
化学
生物
数学
英语
经济学
心理学
社会研究
时尚研究
法律研究随机挑选 C++ 索引
假设我们有一个可能包含重复元素的整数数组，我们必须随机挑选一个给定目标数字的索引。我们可以假设给定的目标数字一定存在于数组中。因此，如果数组如下所示：[1,2,3,3,3]，则 pick(3) 可能随机返回 2、3 或 4。
为了解决这个问题，我们将遵循以下步骤 -
ret := - 1, cnt := 1
对于 i 介于 0 至 v 的大小
如果 v[i] = target，那么
如果随机数模 cnt = 0，那么 ret = i
cnt := cnt + 1
返回 ret
示例 (C++)
让我们看看以下实现以获得更好的理解 -
#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
vector <int> v;
Solution(vector<int>& nums) {
srand(time(NULL));
v = nums;
}
int pick(int target) {
int ret = -1;
int cnt = 1;
for(int i = 0; i < v.size(); i++){
if(v[i] == target){
if(rand() % cnt++ == 0) ret = i;
}
}
return ret;
}
};
main(){
vector<int> v = {1,2,3,3,3};
Solution ob(v);
cout << (ob.pick(3));
}输入
Initialize with [1,2,3,3,3] Call pick(3) to get random index positions
输出
4 3 4 2
更新于： 2020 年 5 月 2 日
640 次浏览
广告