Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution.
Example:
Given nums = [2, 7, 11, 15], target = 9, Because nums[0] + nums[1] = 2 + 7 = 9, return [0, 1].
UPDATE (2016/2/13):
The return format had been changed to zero-based indices. Please read the above updated description carefully.
1.直接暴力解决,用两个for循环依次查找。时间复杂度O(n^2),空间复杂度O(1)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int l=nums.size();
vector<int> a(2);
for(int i=0;i<l;i++){
for(int j=i+1;j<l;j++)
if(nums[j]==target-nums[i])
{
a[0]=i;
a[1]=j;
return a;
}
}
return a;
}
};
2. 用一个无序multimap,之所以用无序是因为无序比有序的快,不用排序。multimap一个键值可以对应多个值。将数组中的整型数和其对应的索引值存储到multimap中,整形数作为键值。然后用一个for循环在multimap中进行查找。时间复杂度O(n),空间复杂度O(n)。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_multimap<int,int> umap;
vector<int> a(2);
for(int i=0;i<nums.size();i++){
umap.insert(pair<int,int>(nums[i],i));
}
for(int i=0;i<nums.size();i++){
int c=target-nums[i];
if(umap.find(c)!=umap.end()&&umap.find(c)->second!=i){//注意查找到的索引值不能与当前索引值相同。
a[0]=i;
a[1]=umap.find(c)->second;
return a;
}
}
return a;
}
};
3.边插入边查找。用一个for循环对数组中的每一个数进行查找,对于当前数,现在multimap中检查有没有满足条件的值,如果有,直接返回。如果没有则插入,这样可以不用判断查找到的数的索引是不是当前数的索引,因为当前数还没有插入到multimap中,所以如果能查找,查找到的数的索引一定与当前数的索引不同。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_multimap<int,int> umap;
vector<int> a(2);
for(int i=0;i<nums.size();i++){
int c=target-nums[i];
if(umap.find(c)!=umap.end()){
a[0]=umap.find(c)->second;
a[1]=i;
return a;
}
umap.insert(pair<int,int>(nums[i],i));
}
return a;
}
};