题目:给定一个整数数组 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处
首先想到就是暴力遍历,时间复杂度为O(n²)
public static int[] twoSum(int[] nums,int target){
int[] label = new int[2];
for(int i=0;i<nums.length;i++){
for(int j=0;j<nums.length;j++){
if(nums[i]+nums[j]==target && i<j){
label[0] = i;
label[1] = j;
}
}
}
return label;
}
然后就想着怎么能减少耗时,在参考他人思路之后`
public static int[] twoSum2(int[] nums,int target){
int[] temp = new int[nums.length]; //存放原始数组
int[] label = new int[2]; //存放结果
for(int i=0;i<nums.length;i++){
temp[i] = nums[i];
}
Arrays.sort(nums); //对数组进行排序
int i = 0;//排序后最小索引
int j = nums.length-1;//排序后最大索引
while(i<j){
//两边夹取值
if(nums[i] + nums[j] == target){
// 存放当前索引的值
label[0] = nums[i];
label[1] = nums[j];
break;
}else if(nums[i] + nums[j] >target){
j--;//
}else{
i++;
}
}
int[] result = new int[2];//第二次bug,当int[] nums = [-18,12,3,0];target = -6时 出错
int count1 = 0;//第四次bug 当int[] nums = [2,5,5,11] target= 10
int count2 = 0;
for(int k = 0;k<temp.length;k++){
if(temp[k] == label[0] && count1 == 0){
result[0] = k;
count1++;//防止重复进入
continue;//第一次bug,当int[] nums = {2,5,5,11}; target = 10,出错
}
if(temp[k] == label[1] && count2 == 0){
result[1] = k;
count2++;
continue;
}
}
Arrays.sort(result);//第3次bug,当 int[]nums = [-1,-2,-3,-4,-5] target = -8
return result;
}
当前方法时间复杂度为O(nlog2N),时间明显缩短。
也可以使用空间换时间,使用HashMap.containsKey()方法;时间复杂度为O(n);
public static int[] twoSum3(int[] nums,int target){
int[] label = new int[2];
HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
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 ){
if(i<map.get(target-nums[i])){
label[0] = i;
label[1] = map.get(target - nums[i]);
}
}
}
return label;
}
时间竟然多了1ms,但是代码比上面的方法少了。