原题链接:https://oj.leetcode.com/problems/two-sum/
撇开暴力破解,这道题有众所周知的两种做法:
1. 先排序,再一前一尾双指针。这种方法要注意的是,排序时需要保存原来的index,这样的话其实是一个有着value和index的结构体基于value的排序。时间复杂度O(n log n)。空间复杂度O(n),因为要保存index。
2. 直接hash。这道题需要注意的是,如果数组有两个相同元素,但是仍然构成唯一解。比如target是6,数组是1,3,3。所以hash的key是int,但是value则是一个vector<int>保存所有的index。网上很多博客都假设数组每个值都是unique的,可能可以通过测试,但是不是solid的解法。时间复杂度O(n),空间复杂度O(n)
class Solution {
public:
vector<int> twoSum(vector<int> &numbers, int target) {
vector<int> results;
unordered_map<int, vector<int> > map;
for (int i = 0; i < numbers.size(); ++i) {
if (map.find(numbers[i]) == map.end()) {
map.insert(make_pair(numbers[i], vector<int> (1, i)));
} else {
map[numbers[i]].push_back(i);
}
}
for (int i = 0; i < numbers.size(); ++i) {
int searched = target - numbers[i];
if (map.find(searched) != map.end()) {
if (searched != numbers[i]) {
results.push_back(i + 1);
results.push_back(map[searched][0] + 1);
break;
} else if (map[searched].size() == 1) {
continue;
} else {
results.push_back(map[searched][0] + 1);
results.push_back(map[searched][1] + 1);
break;
}
}
}
sort(results.begin(), results.end());
return results;
}
};