首页 > 代码库 > [Leetcode] 1 - Two Sum

[Leetcode] 1 - Two Sum

原题链接:https://oj.leetcode.com/problems/two-sum/


撇开暴力破解,这道题有众所周知的两种做法:

1. 先排序,再一前一尾双指针。这种方法要注意的是,排序时需要保存原来的index,这样的话其实是一个有着value和index的结构体基于value的排序。

2. 直接hash。这道题需要注意的是,如果数组有两个相同元素,但是仍然构成唯一解。比如target是6,数组是1,3,3。所以hash的key是int,但是value则是一个vector<int>保存所有的index。网上很多博客都假设数组每个值都是unique的,可能可以通过测试,但是不是solid的解法。

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;
    }
};





[Leetcode] 1 - Two Sum