题目描述
给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。 你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。 示例:
1
2
3
| 给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
|
题解
最初的思路：双重循环，判断2个数之和是否为目标值，如果是则直接找到解，跳出循环；否则，继续循环。
题目代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int>res;
bool flag = false;
for(int i = 0; i < nums.size(); ++i){
if(!flag){
for(int j = i + 1; j < nums.size(); ++j){
if(nums[i] + nums[j] == target){
res.push_back(i);
res.push_back(j);
flag = true;
break;
}
}
}
else{
break;
}
}
return res;
}
};
|
执行结果
优化
有效判断数组中是否存在目标元素，并且能返回下标的快速方法是哈希表 将值及索引添加到哈希表中，建立索引。
代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
| class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int>res;
map<int, int>myMap;
for(int i = 0; i < nums.size(); ++i){
if(myMap.find(nums[i]) != myMap.end()){
res.push_back(myMap[nums[i]]);
res.push_back(i);
break;
}
myMap[target - nums[i]] = i;
}
return res;
}
};
|
执行结果
进一步优化
unordered_map对元素的查询速度要比map快，这里可以用unordered_map代替map。
优化代码如下：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
| class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int>res;
unordered_map<int, int>myMap;
for(int i = 0; i < nums.size(); ++i){
if(myMap.find(nums[i]) != myMap.end()){
res.push_back(myMap[nums[i]]);
res.push_back(i);
break;
}
myMap[target - nums[i]] = i;
}
return res;
}
};
|
优化代码执行结果
