1.题目描述
给定一个整数数组nums和一个整数目标值 target,请你在该数组中找出和为目标值target的那两个整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 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]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
2.解题思路
之前我们已经使用过数据结构unordered_set,它可以帮助我们很简单地解决一些问题,但它只能够记录一个key值,而这里我们需要返回的是对应元素的下标,因此我们需要一种key value数据结构,即记录对应的数值key,并保存其下标value。
这里我们就要用到map数据结构,c++中提供了三种map数据结构如下所示。
3.代码实现
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
std::unordered_map<int, int> map;
for(int i = 0; i < nums.size(); i++)
{
auto iter = map.find(target - nums[i]);
if(iter != map.end())
{
return {iter->second, i};//如果在哈希表中找到对应元素,返回该元素的下标和i
}
map.insert(pair<int, int>(nums[i], i));//将nums中的元素全部映射到map中
}
return {};
}
};