LeetCode1 TwoSum
题目描述:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解答:
解法一:暴力求解
两层循环。
//暴力解法,双重循环。
public static int[] solve(int[] nums, int target) {
int[] result = new int[2];
a:for(int i=0; i<nums.length;i++){
for(int j=i+1; j<nums.length;j++){
if(nums[i]+nums[j]==target) {
result[0]=i;
result[1]=j;
break a;
}
}
}
return result;
}
解法二:利用map
改成查找表方法,当前元素为nums[i],我们只需要找到target-nums[i].
用map,key为数组中的值,value为索引。
为了防止出现:相同元素放入map中会覆盖掉之前的索引,但是结果可能就是这两个元素。
所以我们map只放入当前元素之前的所有元素。
时间复杂度为O(n)
private static int[] solve2(int[] nums, int target){
int[] result= new int[2];
Map<Integer,Integer> map = new HashMap<>();
for (int i=0;i<nums.length;i++){
int second = target-nums[i];
if (map.containsKey(second)){
result[0]=map.get(second);
result[1]= i;
break;
}
map.put(nums[i],i); //如果找不到第二个值,那么就把这个值放进map中。(ps:题目中有说只有一组答案)
}
return result;
}