Day21[26/3/21]T1两数之和
给定一个整数数组 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]
提示:
2 <= nums.length <= 104-109 <= nums[i] <= 109-109 <= target <= 109- 只会存在一个有效答案
进阶:你可以想出一个时间复杂度小于 O(n2) 的算法吗?
解题思路
很简单,你从头到尾遍历每一个元素,检查这个元素的前面有无大小为 target-nums[i] 的元素即可。
然后速度优化在于这个检查,你可以创建一个哈希表用于存储该元素之前的所有元素,因为你只想知道一个具体的元素的存在与否与下标,所以,该元素作为 key 值并用记录其下标作为 value。
(因为哈希表的查找速度是 O(1))
Code
#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;
class Solution
{
public:
vector<int> twoSum(vector<int> &nums, int target)
{
unordered_map<int, int> mp;
for (int i = 0; i < nums.size(); i++)
{
// 如果互补元素存在
if (mp.count(target - nums[i]))
{
return {mp[target - nums[i]], i};
}
else
{
// 否则就存入元素
mp[nums[i]] = i;
}
}
return {-1, -1};
}
};
auto main() -> int
{
vector<int> nums{2, 7, 11, 15};
int target = 9;
Solution sol;
cout << "[";
for (auto index : sol.twoSum(nums, target))
{
cout << index << ',';
}
cout << "]" << endl;
}