两数之和——哈希表解题

题目描述

给定一个整数数组 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]

解题方法

1、暴力法（时间复杂度为O(n^2))

解题思路

        使用暴力遍历法，针对每个元素nums[i]都遍历一遍数组nums，判断nums[i]+nums[j]是否等于target，如果是则记录索引并退出循环。

代码实现

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

复杂度分析

（1）时间复杂度：对于每一个数组元素nums[i] 都需要遍历一次数组nums，则时间复杂度为O(n^2)。

（2）空间复杂度：使用了辅助数组index存储符合条件的索引，则空间复杂度为O(n)。

2、哈希表（时间复杂度为O(n))

解题思路

        上述暴力法的时间主要花费在寻找target-nums[i]上，若使用数组结构进行存储，则每次寻找的时间复杂度就是O(n)（根据索引取值时数组为随机存取，时间复杂度为O(1)，当根据值寻找索引时数组为顺序存取，时间复杂度为O(n)），此时考虑使用哈希表将寻找target-nums[i]的时间复杂度降低至O(1)。

        哈希表的介绍以及STL库中哈希表的使用可以参考以下博客：一文看懂哈希表并学会使用C++ STL 中的哈希表_哈希表有哪些函数-CSDN博客

（1）哈希表键的选择：使用哈希表需要确定哈希表的键（哈希表的键具有唯一性：键值唯一），此处我们在寻找过程中要对比的是数组元素值，因此可以将数组元素值作为键，将元素下标作为值，当确定nums[i]后，以target-nums[i]作为键，在哈希表中寻找该键，如果键不存在则直接返回，反之则以时间复杂度O(1)直接返回对应的下标值。

（2）哈希表键值重复情况讨论：当数组元素值相等时会出现哈希表键值重复（相同数组元素对应不同下标）。

        ①重复元素和其余任意元素之和都不符合条件，则覆盖或者插入失败都不影响最终结果。

        ②重复元素和自身之和可能符合条件，但由于题干中的“每种输入只会对应一个答案”，因此只有当重复元素数为2时，输出结果只有一组，当重复元素数量大于2时，输出结果大于一组。

        ③重复元素和其余元素之和可能符合条件，但由于题干中的“每种输入只会对应一个答案”，如果重复元素存在和除自身以外的元素之和符合条件，则输出的索引就不是只有一组，组数与重复元素的个数相同，因此该种情况不可能出现。

（3）哈希表的建立：如果先遍历一次数组，将数组值作为键，索引作为键值建立哈希表，因为哈希表的键值具有唯一性，当数组值一样时，采用下标添加会产生覆盖，采用insert函数第二次插入无效，因此对于上述第②种情况将无法正确找到其中一个整数。

        通过上述（1）（2）（3）三步分析，建立哈希表的最好方式，就是边遍历数组边建立哈希表，在遍历过程中同时在哈希表中寻找键target-nums[i]，如果没有找到，则将键值对（nums[i]，i）插入到哈希表中，如果找到则通过at()函数返回下标。对于上述①③情况，发生的覆盖或者插入都没有影响，对于情况②，在第一次遇见键nums[i] 时，哈希表中没有对应键值，则直接插入，当第二次遇见时，首先会在哈希表中寻找键nums[i]（nums[i]=target-nums[i]）从而找到第一次插入的键值对，此时就会直接返回结果，不会进行二次插入。

代码实现

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> hmap;
        vector<int> index;
        // 可以使用迭代器iterator接收find函数返回值，也可以使用auto类型接收
        unordered_map<int, int>::iterator iter;
        for (int i = 0; i < nums.size(); i++) {
            iter = hmap.find(target-nums[i]);
            //  auto it = hashtable.find(target - nums[i]);
            // 若没有该键则返回哈希表最后一个元素的下一位置
            // end函数返回一个指向哈希表结尾位置的下一个元素的迭代器
            if(iter != hmap.end()) {
                index.push_back(i);
                index.push_back(hmap.at(target-nums[i]));
            }
            else{
                // 1.使用insert函数进行插入
                hmap.insert({ nums[i],i }); 
                // 2.使用下标进行插入
                // hmap[nums[i]] = i;
            }
        }
        return index;
    }
};

复杂度分析

（1）时间复杂度：在遍历数组的同时建立哈希表查找哈希表，每次查找时间复杂度为O(1)，则最后的时间复杂度为O(n)。

（2）空间复杂度：使用了辅助哈希表，其哈希表长度最长为n，则空间复杂度为O(n)。