LeetCode:Two Sum

题目链接

Given an array of integers, find two numbers such that they add up to a specific target number.

The function twoSum should return indices of the two numbers such that they add up to the target, where index1 must be less than index2. Please note that your returned answers (both index1 and index2) are not zero-based.

You may assume that each input would have exactly one solution.

Input: numbers={2, 7, 11, 15}, target=9
Output: index1=1, index2=2

注意：元素可能有重复

最简单的做法就是一个两重循环，时间复杂度是O(n^2),下面提供两种更高效的解法

算法1：

将数组的数组映射到哈希表，key是元素的值，value是该值在数组中的索引。考虑到数组中元素有重复，我们使用STL中的unordered_multimap, 它可以允许重复的key存在。映射以后，对于数组中的某个元素num，我们只要在哈希表中查找num2 = target-num。需要注意的是在哈希表中找到了num2，并不一定代表找到了题目要求的两个数，比如对于数组2 7 11 15，target = 4，当num = 2时，num2 = target-num = 2，此时num2可以在哈希表中找到，但是num和num2指向的是同一个元素。因此当num2 = num时，在哈希表找到num2的同时，还需要保证哈希表中num2的个数>=2。

该算法时间复杂度为O（n）

class Solution {
public:
    vector<int> twoSum(vector<int> &numbers, int target) {
        int n = numbers.size();
        vector<int> res;
        unordered_multimap<int, int> umap;
        for(int i = 0; i < n; i++)
            umap.insert(make_pair(numbers[i], i));
        for(int i = 0; i < n; i++)
        {
            auto range = umap.equal_range(target - numbers[i]);
            if(range.first != umap.end())//found
            {
                if(numbers[i] != target - numbers[i])
                {
                    auto range2 = umap.equal_range(numbers[i]);
                    res.push_back(min(range.first->second, range2.first->second) + 1);
                    res.push_back(max(range.first->second, range2.first->second) + 1);
                }
                else
                {
                    auto ite = ++(range.first);
                    if(ite != range.second)
                    {
                        auto range2 = umap.equal_range(numbers[i]);
                        res.push_back(min(ite->second, range2.first->second) + 1);
                        res.push_back(max(ite->second, range2.first->second) + 1);
                    }
                }
            }
        }
        return res;
    }
};

算法2：

首先对数组按小到大排序，然后设定两个指针head、tail分别指向排序好的数组的首尾：                                     本文地址

• 如果两个指针对应的元素和等于target，那么找到了
• 如果两个指针对应的元素和小于target，那么需要增加和的大小，则把head指针向后移动
• 如果两个指针对应的元素和大于target，那么需要减少和的大小，则把tail指针向前移动
• head赶上tail指针时，结束

由于本题中需要返回最后找到的数对的索引，因此，排序是我们不移动原来数组的元素，只是把元素的的索引放到一个新的数组，对这个新的索引数组排序

class Solution {
private:
    static vector<int> *numbersCopy;
    static bool cmp(int idx1, int idx2)
    {
        return (*numbersCopy)[idx1] < (*numbersCopy)[idx2];
    }
public:
    vector<int> twoSum(vector<int> &numbers, int target) {
        numbersCopy = &numbers;
        int n = numbers.size();
        vector<int> res;
        vector<int> idx(n);
        for(int i = 0; i < n; i++)
            idx[i] = i;
        sort(idx.begin(), idx.end(), Solution::cmp);
        
        int head = 0, tail = n-1;
        while(head < tail)
        {
            if(numbers[idx[head]] + numbers[idx[tail]] < target)
                head++;
            else if(numbers[idx[head]] + numbers[idx[tail]] > target)
                tail--;
            else //found
            {
                res.push_back(min(idx[head], idx[tail]) + 1);
                res.push_back(max(idx[head], idx[tail]) + 1);
                break;
            }
        }
        
        return res;
    }
};
vector<int> * Solution::numbersCopy = NULL;

 

推荐参考博客：

程序员编程艺术：第五章、寻找满足和为定值的两个或多个数

求和问题总结(leetcode 2Sum, 3Sum, 4Sum, K Sum)

编程之美2.12-快速寻找满足条件的两个数

 

【版权声明】转载请注明出处：http://www.cnblogs.com/TenosDoIt/p/3647244.html