Java基础算法 - 两数之和
作为在力扣刷的第一道题,我在这里记录一下我的解法
两数之和
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并输出他们的数组下标。
假设每种输入只会对应一个答案,不能重复利用这个数组中同样的元素。
枚举法
在数学和计算机科学理论中,一个集的枚举是列出某些有穷序列集的所有成员的程序,或者是一种特定类型对象的计数。这两种类型经常(但不总是)重叠。 枚举在日常生活中很常见,例如表示星期的SUNDAY、MONDAY、TUESDAY、WEDNESDAY、THURSDAY、FRIDAY、SATURDAY就是一个枚举。
听到两数之和,我们最容易想到的就是两数相加,和为目标值,那么可以枚举数组中的每个数值X
,去寻找数值中是否存在target - X
因为不能重复利用数组中相同的数值,所以我们只需要在X
之后寻找target - X
即可,下面是示例代码
int[] nums = new int[] { 3, 8, 6, 9, 15, 24 };
int target = 27; // 目标值
// 不会遍历最后一个元素,因此-1
for (int i = 0; i < nums.length - 1; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[i] + nums[j] == target) {
System.out.println("输出:" + i + "," + j);
return;
}
}
}
但是这种方法效率较低,最坏的情况是数值中任意两个数值都要匹配一次。
下面介绍另一种更优秀的方法。
查表法
所谓查表法就是把事先计算或测得的数据按照一定顺序编制成表格,查表程序的任务就是根据被测参数的值或者中间结果,查出最终所需要的结果。
在遍历的同时,记录一些信息,以省去一层循环,这是以‘空间换时间’的思路。
需要记录已经遍历过的数值和它对应的下标,可以借助查找表实现,下面看详细代码
/**
* 给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。
* 你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
* @param nums 数组
* @param target 目标值
* @return
*/
public static int[] twoSum(int[] nums, int target) {
// 创建一个HashMap
Map<Integer, Integer> hashMap = new HashMap<Integer, Integer>();
// 直接记录第一个数值和下标,省去一次循环
hashMap.put(nums[0], 0);
for (int i = 1; i < nums.length; i++) {
// 目标值减去当前遍历的数值等于已经记录过的数值
// 说明这两个数值符合条件
if (hashMap.containsKey(target - nums[i])) {
// 返回这两个数值的下标
return new int[]{hashMap.get(target - nums[i]), i};
}
// 记录遍历过的数值和它的下标
hashMap.put(nums[i], i);
}
return new int[] {0,0};
}
枚举法效率较低的原因是查找target - X
的时间可能过高,因此这里使用HashMap首先查找是否存在target - X
,如果找到,则我们在查找它的索引,如果不存在,再将X
插入到HashMap,这样可以确保X
不会与自己重复