给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum
需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 0 到 n-1。
注意事项
你可以假设只有一组答案。
样例
给出 numbers = [2, 7, 11, 15]
, target = 9
, 返回 [0, 1]
.
挑战
Either of the following solutions are acceptable:
- O(n) Space, O(nlogn) Time
- O(n) Space, O(n) Time
最好想到的是暴力,套两层循环,O(n^2),没什么意思
1 vector<int> twoSum(vector<int> &numbers, int target) { 2 // write your code here 3 vector<int> res; 4 for(int i=0;i<numbers.size()-1;i++){ 5 for(int j=i+1;j<numbers.size();j++){ 6 if(numbers[i]+numbers[j]==target){ 7 res.push_back(i); 8 res.push_back(j); 9 break; 10 } 11 } 12 } 13 return res; 14 }
怎么优化呢?想一想我们为什么需要套第二层循环?
因为我们需要找到第二个被加数使得两数之和为target
那么找这个数有没有更好的方法?用hashmap就可以,找只需要O(1)
1 vector<int> twoSum(vector<int> &numbers, int target) { 2 // write your code here 3 vector<int> res; 4 map<int, int> hashmap; 5 for(int i=0;i<numbers.size();i++){ 6 if(hashmap.find(target-numbers[i])!=hashmap.end()){ 7 res.push_back(hashmap[target-numbers[i]]); 8 res.push_back(i); 9 return res; 10 } 11 hashmap[numbers[i]]=i; 12 } 13 }
这也是常见的用空间换时间的方法