题目描述
给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路
建立一个unordered_map.
时间复杂度为O(n),空间复杂度为O(n)。
代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
int size = nums.size();
// 建立一个哈希表
unordered_map<int, int> myMap;
for (int i = 0; i < size; i++){
myMap[nums[i]] = i;
}
// 遍历
for (int i = 0; i < size; i++){
int k = target - nums[i];
auto it = myMap.find(k);
if (it != myMap.end() && it->second != i){
res.emplace_back(i);
res.emplace_back(it->second);
break;
}
}
return res;
}
};