题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,
并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
给定代码
class Solution {
public int[] twoSum(int[] nums, int target) {
}
}
题解一
暴力破解,双重循环嵌套。
public int[] twoSum(int[] nums, int target) {
if (nums == null || nums.length == 0) {
return null;
}
for (int i = 0 ; i < nums.length ; i++)
{
for (int j = i + 1 ; j < nums.length ; j++)
{
if (nums[i] + nums[j] == target) {
return new int[] {i, j};
}
}
}
return null;
}
时间复杂度 O ( n 2 ) O(n^2) O(n2),空间复杂度 O ( 1 ) O(1) O(1)。
题解二
只需遍历一次数组,将值作为键、下标作为值存到Map中。
如果检测到target - nums[i]
在Map中已存在,直接将两个下标返回。
public int[] twoSum(int[] nums, int target) {
if (nums == null || nums.length <= 1) {
return null;
}
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0 ; i < nums.length ; i++)
{
int need = target - nums[i];
if (map.containsKey(need)) {
return new int[] {map.get(need), i};
}
map.put(nums[i], i);
}
return null;
}
只需遍历一次数组,而且Map的查询复杂度接近
O
(
1
)
O(1)
O(1),没有影响。
空间取决于Map中的元素数量,最坏情况为n。
时间复杂度
O
(
n
)
O(n)
O(n),空间复杂度
O
(
n
)
O(n)
O(n)。