两数之和

November 28, 2018

两数之和

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

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例: 给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]

答案一:

function twoSum(nums, target) {
  for (let i = 0; i < nums.length; i++) {
    for (let j = i + 1; j < nums.length; j++) {
      if (nums[j] == target - nums[i]) {
        return [i, j]
      }
    }
  }
  throw new error("No two sum solution")
}

复杂度分析:

  • 时间复杂度:O(n^2), 对于每个元素,我们试图通过遍历数组的其余部分来寻找它所对应的目标元素,这将耗费 O(n) 的时间。因此时间复杂度为 O(n^2)

  • 空间复杂度:O(1)

答案二:

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */

var twoSum = function (nums, target) {
  let obj = {}
  let arr = []
  for (let i = 0; i < nums.length; i++) {
    let element = target - nums[i]
    if (nums[i] in obj) {
      arr.push(obj[nums[i]])
      arr.push(i)
      return arr
    } else {
      obj[element] = i
    }
  }
}

复杂度分析:

  • 时间复杂度:O(n), 我们只遍历了包含有 n 个元素的列表一次。在表中进行的每次查找只花费 O(1)的时间。

  • 空间复杂度:O(n), 所需的额外空间取决于哈希表中存储的元素数量,该表最多需要存储 n 个元素。


Profile picture

Written by Jonas who lives and works in ShangHai building useful things. You should follow him on Github