Problem
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.
The return format had been changed to zero-based indices.
Example
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
Algorithm
整理一下题意:给定一个int数组和一个target,在数组中找到两个元素使得它们的和等于target,找到后返回这两个元素的下标。题目的输入确保有且仅有一组解,元素下标从0开始。
最简单的想法当然是直接两次遍历,时间复杂度为O(n^2)。这其中有个小插曲,如果第四行代码写作vector<int> result;
则不能通过,写作vector<int> result(4);
则可以通过。可以猜测后者复杂度比前者略低一些,具体原因尚未在网上找到相关解释,后面查阅了STL的资料再PO上来。
如果想要改进一下代码,可以考虑使用map或者set,时间复杂度为O(nlogn)。利用map建立元素与下标的关系,从而一次遍历完成判断。因为题目规定了有且仅有一组解,所以遍历过程中符合条件即停止,下标组合即为要求输出。
Code
//两次遍历版本,O(n^2),用时799ms
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result(2);
for(int i=0;i<nums.size();i++){
for(int j=i+1;j<nums.size();j++){
if(nums[i]+nums[j]==target){
result[0]=i;
result[1]=j;
}
}
}
return result;
}
};
//map版本,O(nlogn),用时29ms
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> index;
vector<int> result(2);
int i;
for(i=0;i<nums.size();i++){
if(index.count(target-nums[i])!=0){
result[0]=index[target-nums[i]];
result[1]=i;
}
index[nums[i]]=i;
}
return result;
}
};