给定一个整数数组 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]

提示:

LeetCode1. 两数之和(2024秋季每日一题 3)_算法
LeetCode1. 两数之和(2024秋季每日一题 3)_leetcode_02
LeetCode1. 两数之和(2024秋季每日一题 3)_双指针_03
只会存在一个有效答案

进阶: 你可以想出一个时间复杂度小于 LeetCode1. 两数之和(2024秋季每日一题 3)_数据结构_04


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

时间复杂度:LeetCode1. 两数之和(2024秋季每日一题 3)_哈希表_05

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;
    }
};

拓展:也可以考虑用哈希表去优化,用 hash 表记录每个元素的下标(key: 元素值,value:下标),然后扫描一遍,查找 (target - 当前元素) 是否存在于 hash 表中,时间复杂度:LeetCode1. 两数之和(2024秋季每日一题 3)_数据结构_06