题目链接
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]典型的空间换时间,找出一种数据结构存储每个nums[i]对应的下标即可,一旦发现当前的target-nums[i]是存在于这种数据结构中的,就可以返回答案了。
为了降低查询的时间复杂度,我选用的是unordered_map,底层用哈希表实现,查找的时间复杂度为O(1)。
class Solution { public: vector<int> twoSum(vector<int>& nums, int target) { if(nums.size()==0) return vector<int>(); vector<int> ans; unordered_map<int,int> map; for(int i=0;i<nums.size();i++){ if(map.find(target-nums[i])!=map.end()){ ans.push_back(map[target-nums[i]]); ans.push_back(i); return ans; } map[nums[i]]=i; } return ans; } };