题目信息

  • 题目：剑指 Offer 57 - I. 和为s的两个数字

  • 时间： 2020-08-14

  • 题目链接：Leetcode

  • tag： 双指针 哈希表

  • 难易程度：简单

  • 题目描述：

    输入一个递增排序的数组和一个数字s，在数组中查找两个数，使得它们的和正好是s。如果有多对数字的和等于s，则输出任意一对即可。

示例1:

1
2
输入：nums = [2,7,11,15], target = 9
输出：[2,7] 或者 [7,2]

示例2:

1
2
输入：nums = [10,26,30,31,47,60], target = 40
输出：[10,30] 或者 [30,10]

注意

1
2
1. 1 <= nums.length <= 10^5
2. 1 <= nums[i] <= 10^6

解题思路

本题难点

查找和为s的两个数字，性能最优。

具体思路

本题的 nums 是 排序数组 ，因此可使用 双指针法 将空间复杂度降低至 O(1)。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
    public int[] twoSum(int[] nums, int target) {
      //双指针 i , j 分别指向数组 nums 的左右两端 
        int i = 0;
        int j = nums.length - 1;
        while(i < j){
          //计算和 s=nums[i]+nums[j] 
            int sum = nums[i] + nums[j];
          //若 s>target ，则指针 j 向左移动，即执行 j=j−1
            if(sum > target){
                j--;
            }else if(sum < target){//若 s<target ，则指针 i 向右移动，即执行 i=i+1 
                i++;
            }else{//若 s=target ，立即返回数组 [nums[i],nums[j]] 
                return new int[]{nums[i],nums[j]};
            }
        }
        return new int[0];
        
    }
}

复杂度分析：

  • 时间复杂度 O(N) ： N为数组 nums 的长度；双指针共同线性遍历整个数组。
  • 空间复杂度 O(1) ： 变量 i, j 使用常数大小的额外空间。

其他优秀解答

解题思路

哈希表查找，利用 HashMap 可以通过遍历数组找到数字组合，时间和空间复杂度均为 O(N)；

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class Solution {
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer,Integer> map=new HashMap();
        for(int i=0;i<nums.length;i++){
            map.put(target-nums[i],nums[i]);
        }
        for(int i=0;i<nums.length;i++){
        //注意排除当前元素
            if(map.containsKey(nums[i]) && map.get(nums[i])!=i){
                return new int[]{nums[i],map.get(nums[i])};
            }
        }
        return new int[0];
    }
}