给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
Given an array of integers nums and an integer target, return indices of the two numbers such that they add up to target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
You can return the answer in any order.
Example 1:
Input: nums = [2,7,11,15], target = 9
Output: [0,1]
Output: Because nums[0] + nums[1] == 9, we return [0, 1].
Example 2:
Input: nums = [3,2,4], target = 6
Output: [1,2]
Example 3:
Input: nums = [3,3], target = 6
Output: [0,1]
链接:https://leetcode-cn.com/problems/two-sum
解题思路:
O(n)的时间复杂度
先遍历一遍数组,用map容器来存key和value(数组值对应的索引)。
利用在map中查找是常数级的时间复杂度,再次遍历数组中的每个元素nums[i],如果需要两个数之和为target,则数组中另外的那个数肯定为tmp = target - nums[i],我们在map容器中查找这个数就可以了,找到tmp之后,它的index就是map[tmp]。
代码如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> map;
vector<int> result;
int size = nums.size();
for(int i = 0; i < size; i++)
{
map[nums[i]] = i;
}
for(int i = 0; i < size; i++)
{
int tmp = target - nums[i];
if(map.find(tmp) != map.end() && map[tmp] > i)
{
result.push_back(i);
result.push_back(map[tmp]);
break;
}
}
return result;
}
};