[Leetcode]两数之和-哈希+C++
题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解题思路
- 哈希,遍历nums[i]时,判断target-nums[i]是否在哈希表中
- 时间复杂度O(N),空间复杂度O(N)
实现代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
map<int,int> mymap;
int len = nums.size();
for(int i=0;i<len;i++){
int another = target - nums[i];
if(mymap[another]&&mymap[another]!=i+1){//避免同一个元素使用两遍
res.push_back(mymap[another]-1);
res.push_back(i);
}
mymap[nums[i]]=i+1;//处理下标为 0 的情况,mymap[another]=0表示存在但是却不能进入判断
}
return res;
}
};