两数之和算法
题目
给定一个整数数组nums和一个目标值target,在整数数组中找出和目标值的那两个整数,并返回他们的下标。
假设每种输入只对应一个答案,同一个元素不能使用两遍。
输入:nums=[2,7,11,15],target=9
输出: [0,1]
解释:nums[0]+nums[1]=9
所以返回 [0,1]
分析
1.暴力法
这个很好理解,外面遍历一层,里面遍历一层,如果合起来结果为9,就直接返回,其余特殊情况返回两个-1。
时间复杂度 O(n^2)
空间复杂度O(1)
2.哈希法
哈希表可以简单的理解为一对key:value
进行存储
1.创建一个哈希表,key为数组的值,value为该值的下标
2.遍历数组,在哈希表中查找是否有target-num[i]
,如果有就返回。
举个栗子:
输入:nums=[2,7,11,15],target=9
第一次:
Hash表–空的
Key | Value |
---|---|
i=0
target-nums[0]=9-2=7 hash表中没有
第二次:
Hash表更新–存储nums中的第一个
Key | Value |
---|---|
2 | 0 |
i=1
target-nums[1]=9-7=2 hash表中有
返回现在的i—也就是1 还有key对应的值 0
时间复杂度 O(N)
空间复杂度 O(N)
代码
暴力法代码
public class TwoNum {
public int[] twoSum(int[] nums, int target) {
if (nums == null || nums.length == 0) {
return new int[]{-1, -1};
}
int n = nums.length;
for (int i = 0; i < n; i++) {
for (int j = i + 1; j < n; j++) {
if (nums[i] + nums[j] == target) {
return new int[]{i, j};
}
}
}
return new int[]{-1, -1};
}
public void test() {
int[] arr = new int[]{2, 7, 11, 15};
int[] res = twoSum(arr, 9);
//sout快捷键
System.out.println("result-->" + res[0] + " " + res[1]);
}
public static void main(String[] args) {
TwoNum t = new TwoNum();
t.test();
}
}
哈希法代码
public class TwoNum {
public int[] twoSum_2(int[] nums, int target) {
if (nums == null || nums.length == 0) {
return new int[]{-1, -1};
}
//建立哈希表
HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; i++) {
int n = target - nums[i];
if (map.containsKey(n)) {
return new int[]{map.get(n), i};
}
//更新hash表
map.put(nums[i], i);
}
return new int[]{-1, -1};
}
public void test() {
int[] arr = new int[]{2, 7, 11, 15};
int[] res = twoSum_2(arr, 9);
//sout快捷键
System.out.println("result-->" + res[0] + " " + res[1]);
}
public static void main(String[] args) {
TwoNum t = new TwoNum();
t.test();
}
}