Leetcode 刷题1.两数之和

Leetcode1.两数之和

题目描述

给定一个整数数组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)$ 的算法吗？

解题思路

1、暴力算法 $O(n^2)$

遍历一遍数组nums，每次取出nums中的一个数nums[i]，对每个i再遍历一次nums，每次取出nums[j]，如果$nums[i]+nums[j]==target$，则输出i，j。因为题目表示一定有答案，所以不必特判。
但是由于$nums.length<=10^4$，所以$n^2$次最多需要$10^8$次运算，这超出了计算机一秒进行的操作数，所以会超时，因此这个算法无法满足要求。

2、哈希表 $O(n)$

我们可以将数存到哈希表中，具体操作如下：遍历nums，假设当前遍历到的数为x，我们再设置一个数r，使得$r=target-nums[i]$，再从哈希表中查找有没有r，如果有，则将这两个数的下标返回，否则将当前数的数值和下标存到哈希表中。遍历一遍数组的时间复杂度为$O(n)$，从哈希表中查找数据的时间复杂度为$O(1)$，因此该算法时间复杂度为$O(n)$，不会超时。

代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int>head;
        for(int i = 0; i < nums.size(); i++){
            int r = target - nums[i];
            if(head.count(r)) return {head[r],i};
            head[nums[i]] = i;
        }

        return {};
    }
};