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
• 只会存在一个有效答案

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

思路

为了挑战一下，笔者选择尝试使用时间复杂度小于O(n^2)的算法解决该题。

当我们需要查询一个元素是否出现过，或者一个元素是否在集合里的时候，就要首先想到哈希法。本题需要知道已经遍历过的元素及其下标，使用哈希表来存放正合适，而且要让key来存元素，让value来存下标。

该题并不需要key有序，因此使用unordered_map效率更高，算法的时间复杂度为O(n)。

代码

C++版：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> result;
        unordered_map<int,int> num_map; 
        for(int i=0;i<nums.size();i++){
            // 遍历当前元素，并在map中寻找是否有匹配的key
            auto iter=num_map.find(target-nums[i]);
            if(iter!=num_map.end()){
                result.push_back(i);
                result.push_back(iter->second);
            }
            // 如果没找到匹配对，就把访问过的元素和下标加入到map中
            num_map.insert(pair<int,int>(nums[i],i));
        }
        return result;
    }
};

Python版：

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        records = dict()
        # enumerate()函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标。
        for index, value in enumerate(nums):
            if target - value in records:   # 遍历当前元素，并在map中寻找是否有匹配的key
                return [records[target-value],index]
            records[value]=index
        return []