LeetCode题目笔记——1.两数之和

题目描述

这道题可以说是力扣的入坑题了，很经典，好像还是面试的经典题。

给定一个整数数组 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⁴
• -109 <= nums[i] <= 10⁹
• -109 <= target <= 10⁹
• 只会存在一个有效答案

 

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

题目链接

题目难度——简单

方法一：暴力

  最简单的方法就是无脑暴力，虽然看数组长度可能会失败，但平均下来找到答案时两个下标的差距的期望应该不大，可以暴力一试。用两个循环，每次判断两个数相加是否为target。

代码/Python

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        res = []
        n = len(nums)
        for i in range(n):
            for j in range(n):
                if i != j and nums[i] + nums[j] == target:
                    res.append(i)
                    res.append(j)
                    return res
        
        return [-1, -1]    

方法二：哈希表

  我们可以只用一次循环就找到答案，在一次遍历中，对每个x，问题可以变成在数组中找target-x的坐标，如果我们能知道过去遍历过的数的位置i，那么答案就显然是当前位置和i，否则我们继续往前遍历，并记录当前元素的位置。所以我们需要一个字典。

代码/Python

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        res = [-1, -1]
        n = len(nums)
        pos = dict()
        for i in range(n):
            tmp = target - nums[i]
            if tmp in pos:
                res[0], res[1] = i, pos[tmp]
                return res
            else:
                pos[nums[i]] = i
        
        return res    

代码/C++

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res(2);
        unordered_map<int, int> pos;
        int i, tmp;
        for(i = 0; i < nums.size(); i++){
            tmp = target - nums[i];
            if(pos.count(tmp) != 0){
                res[0] = i;
                res[1] = pos[tmp];
                return res;
            }
            else{
                pos[nums[i]] = i;
            }
        }
        return res;
    }
};

总结

  第一种暴力，所以时间是O(N²)，空间是O(1)，第二种遍历一次，时间是O(N)， 使用了字典，所以空间是O(N)。