LeetCode 001. Two Sum

六月ゝ 毕业季﹏ 提交于 2019-12-11 16:07:54

【推荐】2019 Java 开发者跳槽指南.pdf(吐血整理) >>>

题目:

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

好像在《编程之美》里看到过,用hashset.时间复杂度O(N)。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target)
    {
        unordered_set<int> s1(nums.begin(), nums.end());
        for (int index1=0; index1 != nums.size(); ++index1)
        {
            auto pos = s1.find(target-nums[index1]);
            if(pos != s1.end())
            {
                auto pos2 = find(nums.cbegin(), nums.cend(), target-nums[index1]);
                int index2 = pos2-nums.begin();
                if(index1 > index2)
                    swap(index1, index2);
                if(index1 == index2)
                    continue;
                return vector<int>{index1+1, index2+1};
            }
        }
    }
};



标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!