首先看完题目后我首先想到的就是利用哈希表记录每个数对应的索引,然后再通过随机数返回一个随机的下标,非常简单的做法,但是我马上又发现题目警告我 “数组大小可能非常大。 使用太多额外空间的解决方案将不会通过测试。”,害,但是不管了,想都想了先试一下吧,嘿,没想到居然通过了,就是数据惨不忍睹,时间复杂度和空间复杂度都为O(n),以下为代码实现
class Solution
{
unordered_map<int, vector<int>> record;
int len;
public:
Solution(vector<int>& nums)
{
//哈希表记录每个数的所有下标
len = nums.size();
for (int i = 0; i < len; ++i)
{
record[nums[i]].push_back(i);
}
}
int pick(int target)
{
return record[target][rand() % record[target].size()];
}
};
好嘛,看下官方题解,官方题解第一种方法也是哈希表,直接看第二种方法,叫鱼塘抽样,这个方法主要是为了实现题目中要求的每个序列返回的可能性相同,其大致思想为先遍历一次数组,找到所有值为target的数,并用cur来临时记录每个数的下标同时在[0, cur)这个区间取一个数,若该数值为0,则将返回值ans改为该值cur,为什么这样的方法可以保证每一个下标被取到的可能性相等,假设取到下标为i的可能性为f(i),f(i)就应该等于第i次在[0, i)取到了0,而之后都没有再取到0,因此
算法的代码实现很简单
class Solution
{
vector<int>& numss;
int len;
int ans;
public:
Solution(vector<int>& nums):numss(nums)
{
ans = 0;
len = nums.size();
}
int pick(int target)
{
//cur记录target出现次数
int cur = 0;
for (int i = 0; i < len; ++i)
{
if (numss[i] == target)
{
cur++;
if (rand() % cur == 0)
ans = i;
}
}
return ans;
}
};