来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

示例:

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路

暴力法

暴力遍历整个数组,代码如下:

public int[] twoSum(int[] nums, int target) {
  for (int i = 0; i < nums.length - 1; i++) {
    for (int j = i + 1; j < nums.length; j++) {
        if (nums[i] +nums[j] == target) {
          return new int[]{i, j};
        }
    }
  }
  return new int[]{-1, -1};
}

时间复杂度:$O(n^2)$
空间复杂度:$O(1)$

借助map

可以借助map的数据结构存储。
使用key存储差值(target - nums[i])value存储下标i

动图

代码如下:

public int[] twoSum(int[] nums, int target) {
  Map<Integer, Integer> map = new HashMap<>();
  for (int i = 0, len = nums.length; i < len; i++) {
    if (map.containsKey(nums[i])) {
      // key存在,由于key是target和某个数值的差,说明num[i]和map.get(key)加和就是target
      return new int[]{map.get(nums[i]), i};
    } else {
      // 存储差值和下标
      map.put(target - nums[i], i);
    }
  }
  return new int[]{-1,-1};
}

时间复杂度:$O(n)$
空间复杂度:$O(n)$


本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!