题目:https://leetcode-cn.com/problems/two-sum/
方法一:暴力枚举
题目要求找到这两个数,最容易想到的方法就是 两重循环。
/**
* 方法一:暴力枚举
*
* @param nums 数组
* @param target 目标值
* @return 和为目标值的那两个整数的数组下标;若没有符合要求的,则返回 null
*/
public int[] twoSum(int[] nums, int target) {
int len = nums.length;
for (int i = 0; i < len; i++) {
// 这里 j 的下标从 i+1 开始,因为题目要求数组中同一个元素在答案里不能重复出现
for (int j = i + 1; j < len; j++) {
if (nums[i] + nums[j] == target) {
return new int[] { i, j };
}
}
}
return null;
}
时间复杂度: O ( N 2 ) O(N^2) O(N2),其中 N 是数组长度,最坏情况下任意两个数都会匹配一次。
空间复杂度: O ( 1 ) O(1) O(1)。
方法二:哈希表
方法一的时间复杂度较高,原因在于两重循环,那么能否削掉一重循环呢?可以。
借助哈希表,containsKey
方法可以认为是
O
(
1
)
O(1)
O(1) 复杂度,用于替换掉内层循环。
创建哈希表,key 为数组元素值,value 为该元素的下标。遍历数组每个元素 nums[i]
,先查询哈希表中是否存在 target - nums[i]
,然后将 nums[i]
放入哈希表,这样做可以避免“同一个元素在答案里重复出现”。
/**
* 方法二:哈希表
*
* @param nums 数组
* @param target 目标值
* @return 和为目标值的那两个整数的数组下标;若没有符合要求的,则返回 null
*/
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(target - nums[i])) {
return new int[] { map.get(target - nums[i]), i };
}
map.put(nums[i], i);
}
return null;
}
时间复杂度:
O
(
N
)
O(N)
O(N),其中 N 是数组长度。containsKey
方法的复杂度可以认为是
O
(
1
)
O(1)
O(1)。
空间复杂度: O ( N ) O(N) O(N),其中 N 是数组长度。用于哈希表的空间开销。
本题涉及的知识点主要有数组、哈希表等。
本题的优化思路是典型的 空间换时间,这也是极为常用的优化思路。