leetcode:1. 两数之和

题目

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

示例:

  1. 输入:nums = [2,7,11,15], target = 9
  2. 输出:[0,1]
  3. 解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1]
  1. 输入:nums = [3,2,4], target = 6
  2. 输出:[1,2]
  1. 输入:nums = [3,3], target = 6
  2. 输出:[0,1]

解答 & 代码

对于无序数组,一般情况下,我们会首先把数组排序再考虑双指针技巧。TwoSum 启发我们,HashMap 或者 HashSet 也可以帮助我们处理无序数组相关的简单问题

哈希表

  1. class Solution {
  2. public:
  3. vector<int> twoSum(vector<int>& nums, int target) {
  4. // 哈希表,key = 数值,val = 对应的下标
  5. unordered_map<int, int> numIdxMap;
  6. // 遍历数组
  7. for(int i = 0; i < nums.size(); ++i)
  8. {
  9. // 如果哈希表中存在 target - nums[i],则直接返回两个数的下标
  10. if(numIdxMap.find(target - nums[i]) != numIdxMap.end())
  11. return vector<int>{numIdxMap[target - nums[i]], i};
  12. // 将当前的 <元素值, 下标> 存入哈希表
  13. numIdxMap[nums[i]] = i;
  14. }
  15. return vector<int>{-1, -1};
  16. }
  17. };

复杂度分析:

  • 时间复杂度 O(N)
  • 空间复杂度 O(N)

执行结果:

  1. 执行结果:通过
  2. 执行用时:4 ms, 在所有 C++ 提交中击败了 99.40% 的用户
  3. 内存消耗:10.6 MB, 在所有 C++ 提交中击败了 29.57% 的用户