给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
第一眼看到题目就想到了直接用暴力解法,双重for循环,这样其时间复杂度为O(n^2),空间复杂度为O(1)(只用到一个存储计算结果的临时变量):
class Solution {
public int[] twoSum(int[] nums, int target) {
//暴力解法
int temp = 0;
int[] result = new int[2];
for( int i = 0; i<nums.length; ++i){
for( int j = i+1; j<nums.length; ++j){
temp = nums[i]+nums[j];
if(temp == target){
result[0] = i;
result[1] = j;
return result;
}
}
}
return result;
}
}
在看完题解之后发现还有一种解法,利用HashMap查找特性以及以数字作为键值时具有的唯一性(解决了不能是同一数值的问题)。在使用HashMap后程序的时间复杂度变成了O(n),只需要遍历一遍数组即可找到两个符合条件的数字,但这段程序中需要一个长度同样为数组长度的Map来辅助,空间复杂度就变成了O(n),也是一种以空间换时间的做法。
class Solution {
public int[] twoSum(int[] nums, int target) {
//使用HashMap的解法
int temp = 0;
int[] result = new int[2];
Map<Integer,Integer> numsMap = new HashMap<Integer,Integer>();
for( int i = 0; i<nums.length; ++i){
numsMap.put(nums[i],i);
}
for( int i = 0; i<nums.length; ++i){
temp = target - nums[i];
if(numsMap.containsKey(temp) && numsMap.get(temp) != i){
result[0] = i;
result[1] = numsMap.get(temp);
return result;
}
}
return result;
}
}
推而广之:如果我们要取出所有的符合条件的组合呢?
这样我们第二种解法的优势就体现出来了,在第一种解法中,如果要取出所有的组合那么时间的复杂度必然为O(n^2)。