1.问题描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
来源:力扣(LeetCode)链接:https://leetcode.cn/problems/two-sum
2.实例
实例1:
输入:nums = [2,7,11,15],target = 9
输出:[0,1]
解释:因为nums[0]+nums[1]=9,返回他们对应的下标[0,1]
实例2:
输入:nums = [2,3,2],target = 4
输出:[0,2]
解释:因为nums[0]+nums[2]=4,返回他们对应的下标[0,2]
3.解法
使用暴力枚举的方法是最容易想到的,首先固定第一个数,然后在遍历后面的数,如果找到有一个数的和等于target,返回这两个数。如果第一个数和后面的每一个数的和都不满足条件,就换第二个数和后面的数测试,直到找到满足条件的返回。
//使用暴力枚举的方式
public int[] twoSum01(int[] nums, int target) {
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};
}
}
}
//循环结束都还没有找到,说明没有解,返回空值null
return null;
}
这样解的话,时间复杂度就是O(n^2),空间复杂度是O(1),要是数组的nums.length较小的话还无所谓,但是要是数组中的数多起来的话,就很费时间了。 为了节约时间,根据“能量守恒”,那就必须牺牲空间为代价,就看看第二种解法。
上面那种解法我们是拿着一个数num1,在数组中根据下标来查看对应的值是否满足条件,那么如果我们拿着一个数num1,可以直接找target-num1这个数对应的下标就会方便很多。有了这个思路,我们就可以尝试用HashMap来解决这个问题,利用键值对(k-v),以数值为键,以数组下标为值存储在map中,首先从数组nums[]中拿出一个数num1,求出对应的target-num1,利用map.containKey()方法,查看这个数是否在map中,如果存在,返回对应的下标,如果不存在,那么就将num1和对应的下标也存到map中,让map中的数更加丰富。
public static int[] twoSum02(int[] nums, int target){
int leng=nums.length;
HashMap<Integer, Integer> map = new HashMap<>() {};
map.put(nums[0],0);
for (int i = 1; i < leng; i++) {
int log=target-nums[i];
if (map.containsKey(log)){
return new int[] {i,map.get(log)};
}
map.put(nums[i],i);
}
return null;
}
这样的话时间复杂度就降到了O(n),空间复杂度增长O(n)。