Description:
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].分析:
给定一个数组nums和target,求数组中的两个数使得之和为target,并返回其索引。
方法一:暴力枚举法
思路很简单,将所有可能的组合进行判断,返回满足条件组合的索引,其时间复杂度为O(n^n)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ret(2,-1);
int size=nums.size();
for(int i=0;i<size-1;i++){
for(int j=i+1;j<size;j++){
int sum=nums[i]+nums[j];
if(sum==target){
ret[0]=i;
ret[1]=j;
}
}
}
return ret;
}
};
方法二:哈希表
将数组转换为哈希表,通过查找target-nums[i]来确定组合。注意这里将数组的元素作为key,索引作为value,因为我们想要得到的数组索引。这样只需要遍历一次数组,算法的时间复杂度为O(n)。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> ret(2,-1);
//将数组的元素作为Key,索引作为value
unordered_map<int,int> m;
int size=nums.size();
//将整个数组转换为哈希表
for(int i=0;i<size;i++){
m[nums[i]]=i;
}
//查找target-nums[s]是否存在
for(int i=0;i<size;i++){
const int gap=target-nums[i];
if(m.find(gap)!=m.end()&&m[gap]>i){
ret[0]=i;
ret[1]=m[gap];
break;
}
}
return ret;
}
};