LeetCode:1. 两数之和

132 阅读2分钟

本文已参与「新人创作礼」活动, 一起开启掘金创作之路。

1. 两数之和

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

链接:leetcode-cn.com/problems/tw…

思路: 

参考以下链接解法2:

力扣

1.创建一个hashmap(C++ STL中的 hashmap 为 unordered_map<int, int>  hm;)

注意这里的 hm 中:key存数组元素的值value存数组元素的下标

2.从前向后遍历,每遍历一轮就将对应数组的元素值和下标分别存入hm中的key和value中,

并且每轮遍历中都去 hm 中查找是否有与当前 nums[i] 相加之和 为target的 元素sub,

如果有则返回之前遍历时存入元素sub的下标 hm[sub] 与 当前元素下标 i

举例:

nums = [2, 7, 11, 15], target = 9

初始:创建 hm 和 v

第1次循环:i = 0, nums[i] = 2,此时hm中无数据,所以不进入 if 判断,直接将 key=2 和 value=0 存入hm中;

第2次循环:i=1, nums[i]=7,此时hm中已有 key=2 的元素,进入 if 条件,当前元素 7 与之前存入 key=2 的元素和为target.

最后分别返回 key=2 的元素值 hm[2] 和 当前循环的元素 nums[i] 的下标 i 即可,即:[0, 1]

时间复杂度: O(n)  遍历了包含有 n 个元素的列表一次,在表中进行的每次查找只花费 O(1) 的时间

空间复杂度: O(n)  所需的额外空间取决于hashmap中存储的元素数量,该hashmap最多需要存储 n 个元素

// 1.
vector<int> twoSum(vector<int>& nums, int target) {
    unordered_map<int, int> hm;         // key存元素值,value存元素下标
    vector<int> v;                      // 存符合条件的元素下标
    for (int i = 0; i < nums.size(); i++)
    {
        int n = target - nums[i];
        if (hm.find(n) != hm.end())    // 如果nums[i]与hm中的已有元素匹配
        {
             v.push_back(hm[sub]);
             v.push_back(i);
             return v;                 // 返回下标
        }
        hm[nums[i]] = i;               // 不匹配,直接存入hm中
    }
    return v;
}

// 2.类似于1,只是不使用vector存下标,直接返回
vector<int> twoSum(vector<int>& nums, int target) {
    unordered_map<int, int> hm;     // key存元素值,value存元素下标
    for (int i = 0; i < nums.size(); i++)
    {
        int n = target - nums[i];
        if (hm.find(n) != hm.end()) // 如果nums[i]与hm中的已有元素匹配
        {
            return {hm[n], i};      // 返回下标
        }
        hm[nums[i]] = i;            // 不匹配,直接存入hm中 
    }
    return {};
}