给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]`
解题思路
一、暴力法很简单,遍历每个元素 a,并查找是否存在一个值与target−a 相等的目标元素。
class Solution {
public int[] twoSum(int[] nums, int target) {
for(int i=0;i<nums.length;i++){
for(int j=i+1;j<nums.length;j++){
if(nums[j]==target-nums[i]){
return new int[] {i,j};
}
}
}
throw new IllegalArgumentException("No two sum solution");
}
}
这样写提交没有问题但会耗费大量的时间运算。由于需要遍历两次所以时间复杂度是O(n²),空间复杂度较小为:O(1)。
二、题目需要查找索引,然而保持数组中的每个元素与其索引相互对应的最好方法是什么?哈希表!我们自然就想到双列的Map集合,这里我们自然用的HashMap集合。
我们可以在进行迭代并将元素插入到表中的同时,我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在,那我们已经找到了对应解,并立即将其返回。
这样我们就可以节省运算时间,一边给表里插值,一边检查,这样我们就不需要两次遍历。
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer>map=new HashMap<>();
for(int i=0;i<nums.length;i++)
{
int con=target-nums[i];
if(map.containsKey(con)){
return new int[]{map.get(con),i};
}
map.put(nums[i],i);
}
throw new IllegalArgumentException("No two sum solution");
}
}
以空间换取速度的方式,我们可以将查找时间从 O(n) 降低到 O(1)。这样他的时间复杂度就为O(n)×O(1)=O(n)。相对而言因为储存了N次所以它的空间复杂度:O(n)。