输入中可能存在重复值 。
分析:
本题需要返回的是数组下标,因此如果需要使用排序然后双指针的话,需要用到哈希表,但是由于输入中可能存在重复值,因此哈希表的value值必须是vector<int>。
使用双指针求目标值target,利用的是数值大小关系,需要对数组进行排序。
然而原数组下标是答案的话,我们单纯排序是解决不了的。
这里使用一个类似动态规划的哈希表。
一、一层循环+哈希表
一般想法:第一层循环确定一个数nums[i],判断target-nums[i]是否存在,即看所有数 或者 后面的数中存不存在这样的一个数。但是由于数有重复,可能导致一次性做不了。
高级想法:
定义一个集合dp[i],表示包含前i-1个数的集合。第一层循环确定一个数nums[i],判断target-nums[i]是否存在,只需要看dp[i]中是否存在这样的一个数,这样做不具有后效性。
(如果答案是(i,j)j>i,则当判断到j时,nums[i] 一定在dp[j]中,可以找到答案。)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> Hash;
for(int i=0;i<nums.size();++i){
int temp=target-nums[i];
if(Hash.find(temp)!=Hash.end()){
return {i,Hash[temp]};
}
Hash[nums[i]]=i;
}
return {};
}
};
二、扫一遍+挨个看
先扫描一遍,把所有数对应的不同下标存下来,然后一个数一个数看,是否存在他的配偶。(这个时候排序双指针也是可以的)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,vector<int> > Hash;
for(int i=0;i<nums.size();++i){
if(Hash.find(nums[i])==Hash.end()){
Hash[nums[i]]={i};
}else Hash[nums[i]].push_back(i);
}
for(int i=0;i<nums.size();++i){
if(Hash.find(target-nums[i])!=Hash.end()){
if(target!=2*nums[i]||Hash[target-nums[i]].size()==2)
if(target==2*nums[i]){
return Hash[nums[i]];
}else{
return {i,Hash[target-nums[i]][0]};
}
}
}
return {};
}
};
总体来说第二种做法会比较蠢,用形象的例子来看:
假如有n个人排好队,其中只有一对人是同年同月同日生的,请找出这俩人。(当然这里值相等只需要存入一个哈希表,然后遍历哈希表看个数即可,但是为了说明问题请不要这样想。)
第一种方法:从第一个人开始,边看边记录,每次问一个人的生日之后,用其生日看看之前记录的人有没有和ta同年同月同日生的人,一直往后看,一定能看到。如果有多对人是同年同月同日生,也一定能发现。(用哈希表使得这个记录用O(1)就能查到需要查的人是否存在)。
第二种方法:先将所有人记录下来,然后再一个一个人看是否有和ta同年同月同日生的,还需要避免选出来的人就是同一个人。
两个方法都是O(n)的,只是第一个聪明一些,好写一些。