Two Sum

题目来源
给一个数组以及一个targe，求组成这个targe的两个数的索引。
我想的是用哈希表记录下，然后看target-num[i]是不是在哈希表里。代码如下：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> map;
        int n = nums.size();
        vector<int> res;
        for (int i=0; i<n; i++) {
            map[nums[i]] = i;
        }
        for (int i=0; i<n; i++) {
            if (map.count(target-nums[i]) != 0 && i != map[target-nums[i]]) {
                res.push_back(i);
                res.push_back(map[target-nums[i]]);
                return res;
            }
        }
        return res;
    }
};

然后还可以进行优化，变成一次循环就可以，代码如下：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> map;
        int n = nums.size();
        vector<int> res;
        for (int i=0; i<n; i++) {
            if (map.count(target-nums[i]) != 0) {
                res.push_back(i);
                res.push_back(map[target-nums[i]]);
                return res;
            }
            map[nums[i]] = i;
        }
        return res;
    }
};