为了对运行时间复杂度进行优化,我们需要一种更有效的方法来检查数组中是否存在目标元素。如果存在,我们需要找出它的索引。保持数组中的每个元素与其索引相互对应的最好方法是什么?哈希表
通过以空间换取速度的方式,我们可以将查找时间从 O(n) 降低到 O(1)。哈希表正是为此目的而构建的,它支持以 近似 恒定的时间进行快速查找。我用“近似”来描述,是因为一旦出现冲突,查找用时可能会退化到 O(n)。但只要你仔细地挑选哈希函数,在哈希表中进行查找的用时应当被摊销为 O(1)
一个简单的实现使用了两次迭代。在第一次迭代中,我们将每个元素的值和它的索引添加到表中。然后,在第二次迭代中,我们将检查每个元素所对应的目标元素(target - nums[i]target−nums[i])是否存在于表中。注意,该目标元素不能是 nums[i]nums[i] 本身!
#include <iostream>
#include <string>
#include <unordered_map>
#include "math.h"
#include <vector>
#include <unordered_set>
using namespace std;
class Solution{
public:
vector<int> twoSum(const vector<int> &nums,int target){
unordered_map<int,int> mapping;
vector<int> result;
/*第一次迭代吧,把每个严肃的值和他的索引添加到表中*/
for(int i=0;i<nums.size();i++){
mapping[nums[i]]=i; //用哈希表存储每个数对应的下标
}
/*第二次迭代,检查每个元素所对应的目标元素(target-num[i])是否在表中,注意该元素不能为其本身*/
for(int i=0;i<nums.size();i++){
const int gap=target-nums[i];
//遍历哈希表并且不在哈希表中,jiu
if(mapping.find(gap)!=mapping.end()&&mapping[gap]!=i){
result.push_back(i);
result.push_back(mapping[gap]);
break;
}
}
return result;
}
};
int main(int argc, char *argv[])
{
Solution Solu;
int test[] ={2,7,2,1};
vector<int> vec(test,test+sizeof(test)/sizeof(int));
int target=4;
for(int i=0;i<Solu.twoSum(vec,target).size();i++){
cout<<Solu.twoSum(vec,target)[i]<<endl;
}
return 0;
}