lintcode之两数之和

最新推荐文章于 2019-04-26 09:42:09 发布

luckyu1

最新推荐文章于 2019-04-26 09:42:09 发布

阅读量1k

点赞数 1

CC 4.0 BY-SA版权

文章标签：算法 lintcode 面试

本文链接：https://blog.youkuaiyun.com/luckyu1/article/details/51212904

本文介绍了一种解决两数之和问题的有效算法。通过使用哈希表记录元素及其索引，并结合排序与双指针技术，实现了快速查找。特别适用于元素可能重复的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

给一个整数数组，找到两个数使得他们的和等于一个给定的数target。

你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是1到n，不是以0开头。

解答：

hash + 排序 + two pointers,注意数是可能重复的

class Solution {
public:
    /*
     * @param numbers : An array of Integer
     * @param target : target = numbers[index1] + numbers[index2]
     * @return : [index1+1, index2+1] (index1 < index2)
     */
    vector<int> twoSum(vector<int> &nums, int target) {
        // write your code here
        map<int, vector<int>> record;
        for (int i = 0;i < nums.size();++i) {
            record[nums[i]].push_back(i);
        }
        sort(nums.begin(), nums.end());
        int start = 0;
        int end = nums.size() - 1;
        while (start < end) {
            if (nums[start] + nums[end] == target) {
                if (nums[start] == nums[end]) {
                    return {record[nums[start]][0] + 1, record[nums[start]][1] + 1};
                } else {
                    if (record[nums[start]][0] < record[nums[end]][0]) {
                        return {record[nums[start]][0] + 1, record[nums[end]][0] + 1};
                    } else {
                        return {record[nums[end]][0] + 1, record[nums[start]][0] + 1}; 
                    }
                }
            }
                 else if (nums[start] + nums[end] > target) {
                    end--;
                } else {
                    start++;
                }
            }
    }
};