题目描述
给定一个整数数组nums和一个目标值target,请你在该数组中找出和为目标值的两个整数,并返回他们的数组下标。
你可以假设每种输入只对应一个答案。但是你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15],target = 9;因为 nums[0] + nums[1] = 2 + 7 = 9,所以返回 [0, 1]
链接:
https://leetcode-cn.com/problems/two-sum/
解题思路
暴力搜索
一拿到题,首先暴力搜索,通过了。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
for(int i = 0; i < nums.size() - 1; i++) {
for(int j = i + 1; j < nums.size(); j++) {
if(nums[i] + nums[j] == target) {
result.push_back(i);
result.push_back(j);
}
}
}
return result;
}
};
算法的时间复杂度为O(n2),空间复杂度为O(1)。
改进思路
具体思路:关键在于使用哈希表。哈希表开始为空,遍历nums数组,对于第i个元素,如果在哈希表中找不到target与它的差值,则将该元素与其索引的映射存入哈希表中。实际上是用数组中排在后面的元素在哈希表中去寻找排在它前面的和为target的元素。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> res;
//由于unordered_map的速度要比map快,因此选择无序哈希表
unordered_map<int, int> hash;
for(int i = 0; i < nums.size(); i++){
int another = target - nums[i];
if(hash.count(another)){
res = vector<int>{
hash[another], i};
return res;
}
hash[nums[i]] = i;
}
return res;
}
};
哈希表是一种用空间换取时间的数据结构,哈希表的查找时间为O(1),因此算法的时间复杂度为O(n),空间复杂度为O(n)。
注意事项
实现改进算法时,注意在for循环外面也要返回res,否则在LeetCode中执行代码会报错:
solution.cpp: In member function twoSum
Line 6: Char 33: error: control reaches end of non-void function [-Werror=return-type]
unordered_map<int, int> hash;
^~~~
cc1plus: some warnings being treated as errors
这是使用LeetCode要注意的问题。