【Leetcode】【C++】两数之和----简单题解

【Leetcode】【C++】两数之和----简单题解

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

题目描述

给定一个数组nums以及一个目标值target，找出数组中的两个不同位置的数，使得两数之和等于目标值target。

题目思路

容易想到的是暴力双for循环，时间复杂度$O(n^2)$。

题目要求时间复杂度小于$O(n^2)$，所以要另辟蹊径了。

参考官方题解：

1. 使用哈希表(key,value)键值对的特性。针对本题，key=元素的值，value=元素索引。
2. 遍历数组，每遍历一个元素，通过find()函数查找哈希表中是否有target-nums[idx]元素。
   1. 如果有，则直接返回**{当前遍历元素索引idx, 已有元素索引}**。
   2. 若不存在，则将当前元素以**{nums[idx], idx}**的形式记录到哈希表中。
3. 时间复杂度：$O(n)$。只遍历了一次数组。疑问：写入Map中的复杂度是多少？— 网上说最坏$O(n)$，平均$O(1)$。参考博客：https://blog.csdn.net/qq_44918555/article/details/129636776

参考代码

使用 unordered_map，无序映射，即不会自动排序。本题不需要进行排序，减少排序耗时。

// 使用hash表，遍历一遍，遍历是判断target-num是否在表中，如果在，则直接返回{i, idx}即可。
// 如果没有，则将当前遍历的元素以{num, idx}存入hash表
vector<int> twoSum(vector<int>& nums, int target) {
   unordered_map<int, int> hashMap;
   for (int i = 0; i < nums.size(); i++) {
       unordered_map<int, int>::iterator iter = hashMap.find(target - nums[i]); // 查询是否有target-num[i]元素
       if (iter != hashMap.end()) {
           return { iter->second, i };
       }
       // 没有该元素，添加到hashMap中备用
       hashMap[nums[i]]= i;
   }

   return {};
}