[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

标签：

原文地址：http://blog.csdn.net/algorithmengine/article/details/42422743