题目描述
两数之和
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
解题思路
解法一:直观的解法,时间复杂度O(n^2)
遍历每个数,然后从下一个位置开始第二次遍历,查找是否存在一个数与当前数的和等于target,如果找到则返回答案。否则将第一层遍历的下标往后挪一位。如果遍历完整个数组都没有找到符合条件的数据,则返回空
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ans;
if(nums.size() <= 1) return ans;
for(int i = 0; i < nums.size(); i++)
for(int j = i + 1; j < nums.size(); j++)
if(nums[i] + nums[j] == target) {
ans.push_back(i);
ans.push_back(j);
}
return ans;
}
};
解法二:利用空间换取时间,时间复杂度O(n),空间复杂度O(n)
利用hash表将遍历到的数字和下标存起来,每次遍历到某个数时,先判断target在不在hash中,如果存在,则找到一组答案;如果不在则将当前的数和下标存起来。如果遍历完整个数组还没有找到答案,则返回空
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ans;
if(nums.size() <= 1) return ans;
unordered_map<int, int> hash;
for(int i = 0; i < nums.size(); i++) {
if(hash.count(target - nums[i])) return vector<int>{hash[target - nums[i]], i};
hash[nums[i]] = i;
}
return ans;
}
};
网友评论