题目
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
分析
题目让我们找到整数返回下标,很明显只需要两个for循环就可以暴力求解得到结果,但是怎么可以更优化呢?利用map的特性就可以。在Java中用hashmap更简单,这里仅介绍c++,有兴趣可以学一下Java。
解决方案
方法一:暴力法,两个for循环
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int>res;//返回值
for(int i=0;i<nums.size();i++)
{
for(int j=i+1;j<nums.size();j++)
{
if(target==nums[i]+nums[j])
{
res.push_back(i);
res.push_back(j);
return res;//直接跳出俩个for循环,此处可以换成i=nums.size();j=nums.size();
}
}
}
return res;
}
};
方法二:两遍map,说白了就是用了map的函数及特性
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> cur;
vector<int>res;//返回值
int temp;
for(int i=0;i<nums.size();i++)//遍历nums,给map存值
cur[nums[i]]=i;
for(int i=0;i<nums.size();i++)//再次遍历nums,看map中是否存在target-nums[i]
{
temp=target-nums[i];
if(cur.find(temp)!=cur.end()&&cur[temp]!=i)//注意find返回的是迭代器,目标元素不能是 nums[i] 本身
{
res.push_back(i);
res.push_back(cur[temp]);//cur[temp]可以换成cur.find(temp)->second
break;//找到了就退出
}
}
return res;
}
};
方法三:一遍map,将方法二优化了一下
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int>cur;
vector<int>res;//返回值
int temp;
for(int i=0;i<nums.size();i++)
{
temp=target-nums[i];
if(cur.find(temp)!=cur.end())
{
res.push_back(i);
res.push_back(cur[temp]);
return res;//break 两个都可以,区别是前者耗时更短,内存消耗高一点
}else{
cur[nums[i]]=i;
}
}
return res;
}
};