给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
这个题目直接搜索,复杂度是O(n^2),排序后首尾双指针搜索,复杂度是O(nlogn),比较容易想到的方法是先将数组存入到Hash表z中,再遍历数组,每次从哈希表中查询target-nums[i],复杂度是O(1),所以总的时间复杂度是O(n)。空间复杂度是O(n),这个方法还可以优化,因为在遍历数组的过程中,能有两次找到target-nums[i]。所以可以在一次遍历中解决。
Python中哈希表可以用字典实现
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
hash_map = {}
for i,x in enumerate(nums):
if target - x in hash_map:
return [i,hash_map[target-x]]
hash_map[x] = i
return [-1,-1]
下面代码是C++ 中map的使用
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> mp; // 类似python中的字典
/*for(int i=0;i<nums.size();i++)
{
mp[nums[i]] = i; // key是nums[i],value是i
}*/
for(int i=0;i<nums.size();i++)
{
if(mp.find(target-nums[i])!=mp.end()) // 如果找到value为target-nums[i]
{
if(i!=mp[target-nums[i]]) // 排除nums[i]+nums[i]=target的情况
return {i,mp[target-nums[i]]};
}
else
{
mp[nums[i]] = i;
}
}
return {-1,-1};
}
};