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 <= 104
• -109 <= nums[i] <= 109
• -109 <= target <= 109
• 只会存在一个有效答案

思路1

算法：暴力枚举 O(n²)

暴力枚举方法很简单：两重循环枚举下标 i,j，然后判断 nums[i]+nums[j]是否等于 target.

时间复杂度：由于有两重循环，所以复杂度是 O(n²)。

代码

C++代码:

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target)
    {
        vector<int> res;
        for (int i = 0; i < nums.size(); i++)
        {
            for (int j = 0; j < i; j++)
            {
                if (nums[i] + nums[j] == target)
                {
                    res = vector<int>({j, i});
                    break;
                }
            }
            if (res.size() > 0) break;
        }
        return res;
    }
};

思路2

算法：哈希表 O(n)
使用C++中的哈希表——unordered_map<int, int>

循环一遍 nums 数组，在每步循环中我们做两件事：

1. 判断 target-nums[i]是否在哈希表中；
2. 将nums[i]插入哈希表中；

解释：由于数据保证有且仅有一组解，假设是 [i, j] (i<j) ，则我们循环到 j 时，nums[i]一定在哈希表中
，且有 nums[i]+nums[j]==target， 所以我们一定可以找到解。

时间复杂度：由于只扫描一遍，且哈希表的插入和查询操作的复杂度是 O(1)，所以总时间复杂度是 O(n)。

代码

C++代码:

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

python3代码：

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        if not nums or len(nums) < 2:
            return [-1, -1]
        d = {}
        for i, num in enumerate(nums):
            remain = target - num
            if remain in d:
                return[d[remain], i]
            d[num] = i
        return [-1, -1]