1. 两数之和
前言
为了好好学(mian)习(shi),决定还是得刷刷 LeetCode 的题目,这是个人在 LeetCode 做的第一道题,希望通过博客把每道题的思路记录下来。不知道能坚持多久,那就多一篇是一篇吧。对自己的要求:
- 不求最好的解法,只求及格的解法;(毕竟不是为了参加算法竞赛)
- 每篇都包含三个结构:题目、解题思路、解题代码;
祝愿自己能好好努力坚持坚持吧 ~
题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
基本思路
首先我们常规的思路肯定就是两次 for 循环就完事了:
- 第一次 for 循环遍历数组(从 i 开始)
- 第二次 for 循环从 i 的下一个位置 j 开始继续遍历;
- 如果 num[i] + num[j] == target , 则直接返回 new int[] {i, j}
这是一种解法,也是我的第一反应;
但是如果不用 for 循环也可以,我们只需要把索引通过一个 map 记录下来即可,最终解是用了这种方式:
- 创建一个 maps 的结构,key 记录数组值,value 记录索引值;
- for 循环遍历 nums 数组;
- 计算 target 和当前值的差值,判断是否在 maps 中存在;
3.1 如果存在,则返回取出来差值的 index 和当前遍历值的索引,直接返回;
3.2 如果不存在,则将当前值和索引记录到 maps 中;
这种做法会比前一种做法占用多一点内存,但是耗时减少许多,典型的用空间换时间的做法;
实现代码
class Solution {
public int[] twoSum(int[] nums, int target) {
// key为数值,value为数值在 nums 数组中的索引
Map<Integer, Integer> maps = new HashMap<>(nums.length);
Integer subIndex = null;
int subValue = 0;
for (int i = 0; i < nums.length; i++) {
// 得到目标值和当前值的差值
subValue = target - nums[i];
subIndex = maps.get(subValue);
// 如果差值在 map 中已经存在了,则直接返回结果
if (subIndex != null) {
return new int[] {subIndex, i};
}
// 不存在,则将当前值和索引塞到maps中
maps.put(nums[i], i);
}
throw new RuntimeException("No answer");
}
}