【问题】给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
【思路】这个题目使用哈希表可以将算法优化到O(n), 这里面我们只需要遍历一遍哈希表,有一个优化的思路,就是哈希表边创建边查找。其中键值key为数组nums的元素,而value为数组nums的元素的索引。因此当我们遍历到nums[i]时,我们就需要去哈希表中查找target-nums[i],从而得到其索引,因此res将i和hashmap[target-nums[i]]放入数组中!
class Solution { public: vector<int> twoSum(vector<int>& nums, int target) { unordered_map<int, int> hashmap; vector<int> res; for(int i = 0;i < nums.size();i++){ if(hashmap.count(target-nums[i]) > 0){ // 比使用hashmap.find(target-nums[i] != hashmap.end())快一点 res.push_back(hashmap[target-nums[i]]); res.push_back(i); return res; } hashmap[nums[i]] = i; } return res; } };