两数相加(有序/无序) 时间复杂度小于 O(n2)做题心得

原创已于 2022-02-06 18:54:14 修改 · 1.3k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

leetcode

于 2021-11-30 10:37:17 首次发布

博客讨论了如何在有序数组中寻找两个数使得它们的和为目标值。首先介绍了一种使用HashMap的方法，虽然有效但效率不高。接着提出了更优的解法，利用双指针从数组两端开始，根据和与目标值的关系动态调整指针位置，时间复杂度降低到O(n)。这种方法尤其适用于有序数组，展现了双指针在解决此类问题上的优势。

题目如下：
在这里插入图片描述
看到了一个巧妙的解法时间复杂度小于 O(n2)

class Solution {
    public int[] twoSum(int[] nums, int target) {
    int[] arr=new int[2];
        HashMap<Integer,Integer> hashMap=new HashMap<Integer,Integer>();
        //遍历每个数
        for (int i = 0; i < nums.length; i++) {
            //hashmap里都是互补的数，所以只要有一个数等于hashmap里的数就可以输出结果（因为只让输出一种结果）
            if(hashMap.containsKey(nums[i])){
                arr[1]=i;
                arr[0]=hashMap.get(nums[i]);
                return arr;
            }
            //用target减去当前的数，然后把减去后的数存入hashmap
            hashMap.put(target-nums[i],i);
        }

    return arr;
    }
}

那么我们将题目改一下，给的数组中的数都是有序的，那我们该怎么解呢？
在这里插入图片描述

第一种解法： hashmap
跟上面的一样，只不过效率较低

class Solution {
    public int[] twoSum(int[] numbers, int target) {
   HashMap<Integer,Integer> hashMap=new HashMap<>();
   //遍历数组的每个数，不满足条件存进hashmap，满足条件输出
   for(int i=0;i<numbers.length;i++){
       if(hashMap.containsKey(numbers[i])) {
       int a=hashMap.get(numbers[i]);
       int b=i;
       return new int[]{a,b};
       }
       hashMap.put(target-numbers[i],i);
   }
   return new int[0];
    }
}

第二种解法：首位双指针，时间复杂度更低
在这里插入图片描述

解法来自评论区

由于数组递增的特性，所以后面的元素比前面的元素大
可以安排两个指针i,j在两头，记录两个指针的和

    如果和大于target则代表需要缩小指针和,即需要右移j，来缩小指针和

    如果和小于target则代表需要增大指针和,即需要左移i，来增大指针和

class Solution {
   public int[] twoSum(int[] nums, int t) {
        int len = nums.length-1;
        for(int i = 0, j = len; i < j;){
            int sum = nums[i] + nums[j];
            if(sum > t) j--;
            else if(sum < t) i++;
            else return new int[]{i,j};  
        }
        return new int[2];
    }
}