leetcode-1-two sum

#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;
/*
用暴力搜索就要O(n)或者至少是O(nlgn)，似乎会超时。
如果用hash的话就能保证O(n),注意要保存元素的下标，
所以用unordered_map，还要注意两个下标要按从小到大
的顺序。
因为结果只可能有一对下标，所以相同的数字最多有两个，
由于存入unordered_map的时候相同数字的下标会覆盖，
所以比较下标的时候保险就是一个用nums的下标，一个用查找
unordered_map返回的下标。
*/
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> mymap;
vector<int> inds;
for (int i = 0; i < nums.size(); i++) {
mymap.insert(make_pair(nums[i], i));
}
for (int i = 0; i < nums.size(); i++) {
int gap = target - nums[i];
if (mymap.find(gap) != mymap.end()) {
if (i > mymap[gap]) {
inds.push_back(mymap[gap]);
inds.push_back(i);
break;
} else if (i < mymap[gap]){    //一定是两个不同的下标
inds.push_back(i);
inds.push_back(mymap[gap]);
break;
}
}
}
return inds;
}
};
int main(int argc, const char * argv[]) {
Solution s;
vector<int> n = {0,4,3,0};
vector<int> result = s.twoSum(n, 0);
for (int i = 0; i < result.size(); i++) {
cout << result[i] << " ";
}
cout << endl;
return 0;
}

更改（去重）

class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> mymap;
vector<int> inds;
for (int i = 0; i < nums.size(); i++) {
mymap.insert(make_pair(nums[i], i));
}
for (int i = 0; i < nums.size(); i++) {
int gap = target - nums[i];
if (mymap.find(gap) != mymap.end()) {
if (i > mymap[gap]) {
inds.push_back(mymap[gap]);
inds.push_back(i);
break;
}
}
}
return inds;
}
};