简单算法题-两个数相加
题目的大意大概是这个样子的:
给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
一开始的时候我想的就是这个题还不简单 暴力么.
外圈循环定住一个值 内圈循环暴力搜索找到了就return
但是时间复杂度过高
先贴一下暴力的代码吧
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[j] == target - nums[i]) { //target是目标值
return new int[] { i, j };
}
}
}
return null;
时间复杂度为O(n^2) 空间复杂度O(1)
后来我发现有一个超级巧妙的办法 就是用hash表
HashMap<Integer,Integer>();
我们可以让Key 存储 当前元素的值 value 存储对应的数组下标
每遍历一个先看一下 target(目标值) - nums[i] 存不存在这个键
如果存在这个键 返回这个键所对应的value 和 i就可以
如果不存在 那么我们就存到HashMap里
这样一遍遍历就可以
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i];
if (map.containsKey(complement)) {
return new int[] { map.get(complement), i };
}
map.put(nums[i], i);
}
throw new IllegalArgumentException();
}
时间复杂度:O(n)
空间复杂度:O(n)