题目描述
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案,并且你不能使用两次相同的元素。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
提示:
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
- 只会存在一个有效答案
进阶:你可以想出一个时间复杂度小于 O(n^2)
的算法吗?
思路
为了挑战一下,笔者选择尝试使用时间复杂度小于O(n^2)的算法解决该题。
当我们需要查询一个元素是否出现过,或者一个元素是否在集合里的时候,就要首先想到哈希法。本题需要知道已经遍历过的元素及其下标,使用哈希表来存放正合适,而且要让key来存元素,让value来存下标。
该题并不需要key有序,因此使用unordered_map效率更高,算法的时间复杂度为O(n)。
代码
C++版:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
unordered_map<int,int> num_map;
for(int i=0;i<nums.size();i++){
// 遍历当前元素,并在map中寻找是否有匹配的key
auto iter=num_map.find(target-nums[i]);
if(iter!=num_map.end()){
result.push_back(i);
result.push_back(iter->second);
}
// 如果没找到匹配对,就把访问过的元素和下标加入到map中
num_map.insert(pair<int,int>(nums[i],i));
}
return result;
}
};
Python版:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
records = dict()
# enumerate()函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标。
for index, value in enumerate(nums):
if target - value in records: # 遍历当前元素,并在map中寻找是否有匹配的key
return [records[target-value],index]
records[value]=index
return []