题目要求:
1.给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 nums 数组中和为目标值 target 的那两个整数,并返回它们的数组下标。
2.你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
3.你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
解题思路:
对nums数组进行遍历,将数组下标和对应值逐个存储至unordered_map的键值对映射中,利用unordered_map.find()光速查找的特点,这样最多遍历nums数组一遍,即可找到和为target的两个值。
C++代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
//用于保存最后的输出结果
vector<int> answer;
//键值对
unordered_map<int,int> map;
//遍历nums数组
for(int i=0;i<nums.size();i++)
{
//算法核心,找和为目标值的两个数
if(map.find(target-nums[i])!=map.end())
{
//map.find(target-nums[i])返回结果是一个迭代器,里面->first保存的key值,->second保存的是value值
answer.push_back(map.find(target-nums[i])->second);
answer.push_back(i);
return answer;
}
map.insert({nums[i],i});
}
return answer;
}
};
运行结果:
上面是map的运行结果,下面是unordered_map的运行结果,可以看出,unordered_map用空间换时间的效果更好。