LeetCode：1. 两数之和

1. 两数之和

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

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

 

示例:

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

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

链接：https://leetcode-cn.com/problems/two-sum

 

思路： 

参考以下链接解法2：

https://leetcode-cn.com/problems/two-sum/solution/liang-shu-zhi-he-cshi-xian-liang-chong-jie-fa-bao-/

 

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 {};
}