给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
解法1:这道题可以通过两层循环来搞定,但是时间复杂度比较高,最坏情况为o(n^2),代码如下
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int n=nums.size();
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
if(nums[i]+nums[j]==target)
{
return {i,j};
}
}
}
}
};
解法二:设计到查找问题,我们可以想到用哈希表来解决,查找速率为线性,我们可以建立数组的下标和值作为对应关系,然后后用target从数组头部开始减,看值是否存在哈希表内,这样时间复杂度为o(n)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> mapping;
for(int i=0;i<nums.size();i++)
{
mapping[nums[i]]=i; //key值为数组值,value为数组下标
}
vector<int> result;
for(int i=0;i<nums.size();i++)
{
int gap=target-nums[i];
if(mapping.find(gap)!=mapping.end()&&mapping[gap]>i) //查找
{
result.push_back(i);
result.push_back(mapping[gap]);
break;
}
}
return result;
}
};
注意:在查找的过程中我使用的是find,find的返回值是查找到值得位置,如果没有查找到,为end(),这里也可以使用count
count是如果查找到返回1,否则返回0,所以mapping.find(gap)!=mapping.end() 可以替换为mapping.count(t)
https://blog.csdn.net/linfeng886/article/details/79772348 python的解法三 用c++实现
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> vec;
unordered_map<int,int> res;
for(int i=0;i<nums.size();i++){
int x=target-nums[i];
if(res.count(nums[i])){
vec.push_back(res[nums[i]]);
vec.push_back(i);
}
else{
res[x]=i;
}
}
return vec;
}
};