小知识,大挑战!本文正在参与「程序员必备小知识」创作活动
本文已参与「掘力星计划」,赢取创作大礼包,挑战创作激励金。
题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9 输出:[0,1] 解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。 示例 2:
输入:nums = [3,2,4], target = 6 输出:[1,2] 示例 3:
输入:nums = [3,3], target = 6 输出:[0,1]
解题思路
利用Map
的key-value结构,将当前值所需的值和当前值的下标存储在map
中,在循环数组的过程中,一边判断当前值时候是其他值所需的,一边将当前值所需的数值和下标放入map
。
构造Map
集合,Map
的key为目标值target - 当前值
, value为当前值的下标。这样在循环中可以先判断当前值
是否可以和前面的值
构成目标值target
。如何当前值
可以和前面的值
构成目标值target
,那么Map
中那个值的下标和当前值的下标就是要返回的答案。
示例代码
public int[] twoSum(int[] nums, int target) {
// key: 当前值需要的数值(target - 当前值)
// value: 当前值的下标
Map<Integer, Integer> map = new HashMap<>();
int len = nums.length;
for (int i = 0; i < len; i++) {
// 判断当前值是否是前面所需要的
// 如果map可以get到值,说明当前值是是前面所需要的
Integer surplus = map.get(nums[i]);
if (surplus != null && i != surplus){
return new int[]{i, surplus};
}
// 如果当前值不是前面所需要的,那么把当前值需要的数值和下标放入map
map.put(target - nums[i], i);
}
return null;
}
LeetCode运行结果