[Leetcode] 1 - Two Sum

来源:互联网 发布:天才 知乎 编辑:程序博客网 时间:2024/03/28 21:56

原题链接: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;    }};





0 0
原创粉丝点击