两数之和
问题描述:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
暴力法求解:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
for(int i = 0 ; i < nums.size() ; i++){
for(int j = i+1 ; j<nums.size() ; j++){
if(nums[i] + nums[j] == target){
return {i,j};
}
}
}
return {-1,-1};
}
};
时间复杂度为:O(n^2)
空间复杂度为:O(1)
上面的代码size()函数返回vector向量的元素个数
哈希表求解:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> numPair; // 增加了额外的空间复杂度
for(int i = 0 ; i < nums.size() ; i++){
numPair.insert(pair<int,int>(nums[i],i));
}
for(int i = 0 ; i < nums.size() ; i++){
// 第一个条件是判断是否存在另一个被加数
// 第二个条件是排除相同索引的值
if(numPair.count(target-nums[i])==1 && numPair[target - nums[i]] != i){
return {i,numPair[target - nums[i]]};
}
}
return {-1,-1};
}
};
时间复杂度为:O(n)
空间复杂度为:O(n)
上面的insert()为map中添加元素的函数,count()为map中元素是否存在的函数,返回值只有0或者1,0代表不存在,1代表存在