LeetCode1. 两数之和（2024秋季每日一题 3）

sweetheart7-7

于 2024-08-26 15:08:42 发布

阅读量751

点赞数 10

分类专栏：秋季每日一题2024 文章标签：算法 leetcode 数据结构双指针哈希表

本文链接：https://blog.csdn.net/qq_46456049/article/details/141564401

版权

秋季每日一题2024 专栏收录该内容

32 篇文章 0 订阅

订阅专栏

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值 target 的那两个整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案，并且你不能使用两次相同的元素。

你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

提示：

$2 <= nums.length <= 10^4$
$10^9 <= nums[i] <= 10^9$
$10^9 <= target <= 10^9$
只会存在一个有效答案

进阶： 你可以想出一个时间复杂度小于 $O(n^2)$ 的算法吗？

思路：先对数组进行排序，排序之前，用 pair 记下原始下标，对排序后的数组，用双指针找出对应的两个数

时间复杂度： $排序 O (Nl o g N) + 双指针 O (N)$

class Solution {
public:
    vector<int> twoSum(vector<int>& a, int target) {
        vector<int> res;
        vector<pair<int, int>> nums;
        for(int i = 0; i < a.size(); i++){
            nums.push_back({a[i], i});
        }
        sort(nums.begin(), nums.end());

        for(int i = 0, j = nums.size() - 1; i < nums.size(); i++){
            while(j >= 0 && (nums[i].first + nums[j].first) > target) j--;
            if(nums[i].first + nums[j].first == target){
                res.push_back(nums[i].second);
                res.push_back(nums[j].second);
                break;
            }
        }
        return res;
    }
};