[Leetcode] 1 - Two Sum

最新推荐文章于 2021-12-23 23:38:17 发布

原创最新推荐文章于 2021-12-23 23:38:17 发布 · 680 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

135 篇文章

订阅专栏

本文详细介绍了解决LeetCode中两数之和问题的两种常见方法：排序后使用双指针和直接使用哈希表。讨论了每种方法的时间复杂度、空间复杂度，并提供了代码实现。

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