两数之和题解
背景
link
https://leetcode.cn/problems/two-sum/description/
本题是leetcode题库的第一题,作为一道入门级的题目,在解法上也是很有意思的,本文将尝试给出几种常见解法。
题目
给定一个整数数组 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]
解法
- 两数之和,时间复杂度有限,考虑直接两层for循环解决问题,
时间复杂度O(n^2)。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int n = nums.size();
for (int i = 0; i < n - 1; i++) {
for (int j = i + 1; j < n; j++) {
if (nums[i] + nums[j] == target) {
return {i, j};
}
}
}
return {};
}
};
- 作为一个leetcoder,有时候官方给出的时间和内存的排名还是能激起优化的欲望的。显然,优化的常规思路就是降复杂度,两层for循环如果能优化成一层,那么优化效果显而易见。我们每次选择一个数nums[i],所要求得的另一个值则可以表示为target - nums[i]。解法1也正是运用了这种思想,不过它的选择是每次重新遍历一遍,这层的复杂度是O(n), 有没有一种方法能降一降呢?当然在我们选中一个值的同时,尝试通过数据结构的记录而非遍历去查找另一个值是个很好的思路。红黑树是一个不错的结构,可惜的是平均查找复杂度是O(lgn)。幸好,C++还提供了unordered_map, 不考虑数据的有效性的情况下能够实现O(1)的复杂度,这样我们的算法就能优化到O(n),大功告成。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> value;
const int n = nums.size();
for (int i = 0; i < n; i++) {
int k = target - nums[i];
if (value.count(k) != 0) {
return {i, value[k]};
} else {
value[nums[i]] = i;
}
}
return {};
}
};