思路
暴力枚举的话时间复杂度是
O
(
n
²
)
O(n²)
O(n²),既然大量时间都花费在寻找target - x上,那么就要降低查找的时间。使用unordered_map
可以将查找的时间复杂度降到
O
(
1
)
O(1)
O(1),这样的话只需遍历一次数组,时间复杂度就可以降到
O
(
n
)
O(n)
O(n)。
代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> nums_map;
// 将数组的数字和下标录入map
for (int i = 0; i < nums.size(); i++)
nums_map.emplace(nums[i], i);
// 遍历查找
for (int x : nums) {
if (nums_map.find(target - x) != nums_map.end())
return {nums_map[x], nums_map[target - x]};
}
return {};
}
};
但是这个出现了错误解答。
错误在于可能会一个元素重复出现,如[3, 2, 4], 6 就会输出[0, 0]。
为了避免这个问题,需要在每次查找target - x时都不要查找x本身。那么如何在map里不查找一个键呢?只要这个map里没有这个键,自然就不会查找到了。
因此可以改变下思路,将键值对的插入动作放到判断相加之和的后面。即:先遍历数组判断有无target - x,如果没有再将其插入map。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> nums_map;
for (int i = 0; i < nums.size(); i++) {
if (nums_map.find(target - nums[i]) != nums_map.end())
return {i, nums_map[target - nums[i]]};
nums_map.emplace(nums[i], i);
}
return {};
}
};