给定一个整型数组nums和一个目标值target,如果数组中两个元素的和等于目标值,则返回对应的下标,假设只存在一个解
例子:
Input: nums = [2, 7, 11, 15], target = 9
Output: [0, 1]
解题思路:
首先,将数组转存至map,值存为key,下标存为value
然后,从一端访问,tmp = target - nums[i],然后同过map.find函数查找是否存在tmp键,如果存在,则返回i和tmp对应的值
实现:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> map_nums;
vector<int> result;
for(int i = 0;i < nums.size();i++)
{
map_nums.insert(make_pair(nums[i],i));
}
for(int i = 0;i < nums.size();i++)
{
int complement = target - nums[i];
if(map_nums.find(complement) != map_nums.end() && map_nums[complement]!= i)
{
result.push_back(i);
result.push_back(map_nums[complement]);
return result;
}
}
}
在LeetCode上看到更加快速的实现方法,去掉上面解题思路的第一步,在第二步中去添加元素
实现:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> map_nums;
vector<int> result;
for(int i = 0;i < nums.size();i++)
{
int complement = target - nums[i];
if(map_nums.find(complement) != map_nums.end())
{
result.push_back(i);
result.push_back(map_nums[complement]);
return result;
}
map_nums.insert(make_pair(nums[i],i));
}
}