刚学完数据结构,刷下leetcode练练手。
题目:给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
上来直接二话不说暴力解决两个for循环嵌套搞定
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] temp=new int[2];
boolean flag=false;
for(int i=0;i<nums.length;i++){
for(int j=i+1;j<nums.length;j++){
if(nums[i]+nums[j]==target) {
temp[0]=i;
temp[1]=j;
flag=true;
break;
}
if(flag==true)
break;
}
}
return temp;
}
}
看了人家的解析发现原来用hashmap降低时间复杂度,
思路:把数组中的元素全部加到hashmap中key=值,value=下标,然后遍历数组,在hashmap找target-num[i]的元素。由于hashmap查找时间复杂度为O(1),所以最终的时间复杂度为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.get(target-nums[i])!=null&&map.get(target-nums[i])!=i) {
//map若可以找到target-nums[i],并且该值和nums[i]不是同一个值
return new int[] {i,map.get(target-nums[i])};
}
}
throw new IllegalArgumentException("No two sum solution");
}