当前位置:   article > 正文

数据结构篇-04:哈希表解决TwoSum问题

数据结构篇-04:哈希表解决TwoSum问题

本文对应 力扣高频100 中的“1、两数之和”


对于“两数之和”问题,我们常用的方法是暴力遍历 或者 双指针方法。但是后者只有在数组有序的情况下才能使用。

当数组无序时,使用哈希表是最好的选择。通过使用哈希表来记录元素,可以让我们在时间复杂度O(1)中获得目标元素

接下来我会逐一演示以上三种方法的代码示例

哈希表

寻找能与当前元素 nums[i] 相加为目标值 target 的元素

  1. int[] twoSum(int[] nums,int target) {
  2. int n = nums.length;
  3. index<Integer,Integer> index = new HashMap<>();
  4. //将数组元素构建哈希表
  5. for(int i = 0;i < n;i++){
  6. index.put(nums[i],i);
  7. }
  8. //寻找目标元素
  9. for(int i = 0;i < n;i++){
  10. int other = target - nums[i];
  11. //当哈希表中有这个元素且这个元素不是nums[i]本身时
  12. if(index.containsKey(other) && index.get(other) != i){
  13. return new int[]{i,index.get(other)};
  14. }
  15. }
  16. return new int[]{-1,-1};
  17. }

暴力遍历

  1. int[] twoSum(int[] nums,int target){
  2. for(int i = 0;i < nums.length;i++){
  3. for(int j = i + 1;j < nums.length;j++){
  4. if(nums[j] == target - nums[i]){
  5. return new int[]{i,j};
  6. }
  7. }
  8. }
  9. return new int[]{-1,-1};
  10. }

双指针

对于双指针的使用,后面谈到nSum问题时我会详细讲述

  1. int[] twoSum(int[] nums, int target) {
  2. int left = 0, right = nums.length - 1;
  3. while (left < right) {
  4. int sum = nums[left] + nums[right];
  5. if (sum == target) {
  6. return new int[]{left, right};
  7. } else if (sum < target) {
  8. left++; //sum 大一点
  9. } else if (sum > target) {
  10. right--; //sum 小一点
  11. }
  12. }
  13. // 不存在这样两个数
  14. return new int[]{-1, -1};
  15. }

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/思考机器3/article/detail/61299
推荐阅读
相关标签
  

闽ICP备14008679号