题目描述:给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路 参考网上的解法
用unordered_map解决 ,首先定义一个unordered_map<int> hash;
遍历nums数组里的数据,看target与nums[i]的差值在 hash数组中是否存在(用count()方法),如果不存在,就把nums[i]和其下标存在hash中(hash[nums[i]]=i),如果存在,说明正在遍历的nums[i]与前面的nums[i]之和等于target,
此时返回 return {hash[nums[i]-target],i}
1 #include<iostream> 2 #include<unordered_map> 3 #include<vector> 4 #include<cmath> 5 using namespace std; 6 vector<int> twoSum(vector<int>& nums, int target) { 7 unordered_map<int, int> hash; 8 for(int i = 0; i < nums.size(); i++){ 9 if(hash.count(target - nums[i])) 10 return {hash[target - nums[i]], i}; 11 hash[nums[i]] = i; 12 } 13 return {-1, -1}; 14 } 15 int main(void){ 16 vector<int> nums(4); 17 vector<int> res(2); 18 int target; 19 for(int i=0;i<nums.size();++i) 20 cin>>nums[i]; 21 cin>>target; 22 res=twoSum(nums,target); 23 for(int i=0;i<res.size();++i) 24 cout<<res[i]; 25 return 0; 26 }
注意:1、使用unordered_map必须在头文件加上<unordered_map>,unordered_map里的键唯一,如果再插入相同的键,再插入的键会把原来的键覆盖掉。