【算法-Java实现】两数之和
一.问题描述:
给定一个整数数组nums和一个目标值target,找出该数组中和为目标值的那两个整数,并返回他们的数组下标。
二.问题解答:
解法1:暴力法
双层for循环遍历,判断两个数相加是否为target即可
第二个数只能是第一个数后面的数,所以第二层for循环从下标i+1开始
代码如下
class Solution {
public 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[i]+nums[j]==target){
return new int[]{
i,j};
}
}
}
return new int[0];
}
}
算法分析:
时间复杂度:O(N的平方)
空间复杂度:O(1)
解法2:哈希表
创建一个HashMap,键为nums[i],值为i,这样就可以通过get()方法根据键获取值
将数组的nums[i]和i作为键-值对,使用put()方法依次存入HashMap中。
for循环遍历时,int otherNum=target-nums[i],判断otherNum是否存在于HashMap的键中,若存在返回对应数组下标即可。
代码如下
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> map=new HashMap<>();
for(int i=0;i<nums.length;i++){
int otherNum=target-nums[i];
if(map.containsKey(otherNum)){
return new int[] {
map.get(otherNum),i};
}
map.put(nums[i],i);
}
return new int[0];
}
}
注意:map.put(nums[i],i)这个方法要在if判断之后
如果测试用例为nums=[3,2,4],target=6,则会返回[0,0],不满足结果[1,2]
因为若先put()再if判断,判断时刚好otherNum=6-3=3,就会返回两个0下标,也就是造成同一个数使用两遍的情况,这是不符合题目条件的。
算法分析
时间复杂度:O(N): for循环遍历,N为数组的元素个数
空间复杂度:O(N):创建哈希表来存放了数组的值,N为数组的元素个数