力扣刷题之「两数之和」

leetcode 第一题 就是两数之和:

给定一个整数数组 nums 和一个目标值 target,请在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。可以假设每种输入只会对应一个答案。但是数组中同一个元素不能使用两遍。示例:

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

刚看此题时,第一反应就是使用双重循环进行遍历:

1
2
3
4
5
6
7
8
9
10
// 双重循环最直观
var twoSum = function (nums, target) {
for (let i = 0; i < nums.length; i++) {
for (let j = i + 1; j < nums.length; j++) {
if (nums[i] + nums[j] === target) {
return [i, j]
}
}
}
}

虽然解法没有问题,但是时间复杂度是 O(n²),更为优雅的解法是定义一个对象存储目标对象,以空间换时间:

1
2
3
4
5
6
7
8
9
10
11
12
13
// 以空间换时间
var twoSum = function (nums, target) {
const obj = {} // 用于记录想找的那个数字
for (let i = 0; i < nums.length; i++) {
const num = nums[i]
// 如果目标在,就直接返回下标,否则设置新目标
if (num in obj) {
return [obj[num], i]
} else {
obj[target - num] = i
}
}
}

虽然此题难度系数是简单,但是思想却非常巧妙,双重循环解法和空间换时间解法可能就是普通码农和高级码农的区别,既然条条大路通罗马,为什么不走最近的那一条呢!

一起刷题,做个聪明人!