leetcode 1
链接: 力扣 。
题目:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
输入:nums = [2, 7, 11, 15], target = 9
输出:[0, 1]
解释:nums[0] + nums[1] = 2 + 7 = 9
思路:
最直接想到的方法是暴力双循环,时间复杂度为O(n^2)。
更优的方法是使用hash表来做。对于本题应该选择的容器为map,因为我们需要成对的值<key, value>。而STL中map有三种:map、multimap、unordered_map。map和multimap底层是红黑树,会对元素按键值排序。我们并不需要排序,故选择unordered_map。同时,unordered_map底层是哈希表,查找效率高于红黑树。
先建立一个unoredered_map<key, value>容器,元素的值为key,元素的下标为value,这样做是为了方便查找元素。然后遍历数组nums,对于遇到的每个元素nums[i],查找在哈希表中是否有target-nums[i],如果存在,说明我们找到满足要求的两个数,输出当前的nums[i]和哈希表中查找到的那个数;如果不存在,就将nums[i]加入哈希表,继续向后遍历。
参考代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> hashMap;
for (int i = 0; i < nums.size(); i++) {
auto it = hashMap.find(target - nums[i]);
if (it == hashMap.end()) {
hashMap.insert(pair<int, int>(nums[i], i));
}
else {
return {it->second, i};
}
}
return {};
}
};