LeetCode第一题
题目描述:给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
解决方法
- 暴力破解:遍历数组,时间复杂度O( n 2 n^2 n2),空间复杂度O(1)。
public static 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};
}
}
}
return null;
}
- 两遍哈希表:先报key-value put进HashMap,再遍历数组。时间复杂度O(n),空间复杂度O(n)。
public static int[] twoSum(int[] nums, int target) {
HashMap<Integer,Integer> map = new HashMap();
for(int i= 0;i<nums.length;++i)
{
map.put(nums[i],i);
}
for(int i=0;i<nums.length;++i){
if(map.containsKey(target-nums[i])&&map.get(target - nums[i])!=i){
return new int[]{i,map.get(target-nums[i])};
}
}
return null;
}
- 一遍哈希表:一边put key-value,一边判断。时间复杂度O(n),空间复杂度O(n)。
这里有个坑,HashMap的key值不能相同,如果先put再判断,遇到有相同两个数的和等于target(例如:[3,3],target:6),因为后面put的值会覆盖前面put的值,所以无法得到正确的结果。
错误代码如下:
public static int[] twoSum(int[] nums, int target) {
HashMap<Integer,Integer> map = new HashMap();
for(int i= 0;i<nums.length;++i)
{
map.put(nums[i],i);
if(map.containsKey(target - nums[i]) && map.get(target -nums[i])!=i){
return new int[]{i,map.get(target - nums[i])};
}
}
return null;
}
正确代码:
public static int[] twoSum(int[] nums, int target) {
HashMap<Integer,Integer> map = new HashMap();
for(int i= 0;i<nums.length;++i)
{
if(map.containsKey(target - nums[i])){
return new int[]{i,map.get(target - nums[i])};
}
map.put(nums[i],i);
}
return null;
}