题目
来源:LeetCode.
给定一个整数数组 nums 和一个整数目标值 target,
请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。
但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
看到这个问题我首先想到的方法就是暴力题解,枚举出所有的结果。
接下来看一下解题思路:
思路一(暴力求解):
用双层for循环遍历数组,判断两个数加起来是否等于target,如果等于返回对应的下标。
代码示例:
public int[] twoSum1(int[] nums, int target) {
if(nums.length < 2) {
return null;
}
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[0];
}
总结
这样做时间复杂度 O(n^2),最坏情况下,数组遍历完才找到。
空间复杂度 O(1)
思路二(哈希表)
采用空间换时间的方式来降低时间复杂度;
通过哈希表存放,数组值和下标,
* 遍历数组,先判断目标值减去当前值是否存在哈希表中,
* 若存在,就返回当前下标和另一个数的下标,
* 若不存在,就将当前元素存入哈希表中
代码示例:
public int[] twoSum2(int[] nums, int target) {
if(nums.length < 2) {
return null;
}
Map<Integer, Integer> map = new HashMap<>();
for(int i = 0; i < nums.length; ++i) {
int last = target - nums[i];
if(map.containsKey(last)) {
return new int[]{i, map.get(last)};
}
map.put(nums[i], i);
}
return new int[0];
}
总结
这样只遍历了一次数组,所以时间复杂度 O(n),
因为用哈希表存放数据,最坏会存放n - 1个元素,所以空间复杂度 O(n)