给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你能够假设每种输入只会对应一个答案。可是,你不能重复利用这个数组中一样的元素。java
示例:数组
给定 nums = [2, 7, 11, 15], target = 9 由于 nums[0] + nums[1] = 2 + 7 = 9 因此返回 [0, 1]
<!--more-->spa
target
(固然这估计过不了关)HashMap
有个get
方法效率很高,那么首先把数组中全部元素put
到map
中,将target - nums[i]
做为key,索引index
做为value
,那么再写一个循环去遍历数组时候,判断起来就很方便了。map.get(nums[i])
,便可拿到对应的value
,当value
不为null
的时候证实知足了nums[i] + nums[j] = target
,也就是nums[i] = target - nums[j]
。如上面示例的话,map中会存放的是map:{7:0, 2:1, -2:2, -6:3}
int b = map.get(nums[i])
不为空 且b != i
时<font color=grey size=2>(由于当nums[0]==3
,target==6
就会获得[0,0],很显然这是错误的,题目中提到了不能重复利用这个数组中一样的元素)</font>,便可break
,[i,b]
就是答案put
到map
中(key,value
同上),循环过程当中再去判断是否map
已经contains
这个nums[i]
了,若是包含则知足了nums[i] = target - nums[j]
,便可以break
上面都将 $ O(n^2)->O(n) $code
package com.dasnnj.practice.simple; import java.util.Arrays; import java.util.HashMap; import java.util.Map; /** * Description <p> TODO: * 给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。 * 你能够假设每种输入只会对应一个答案。可是,你不能重复利用这个数组中一样的元素。 * <p> * 示例: * <p> * 给定 nums = [2, 7, 11, 15], target = 9 * <p> * 由于 nums[0] + nums[1] = 2 + 7 = 9 * 因此返回 [0, 1]</p> * * @author DASNNJ <a href="mailto:dasnnj@outlook.com"> dasnnj@outlook.com </a> * @date 2019-04-28 15:38 */ public class TwoSum { static int[] twoSum(int[] nums, int target) { //One /*//key为target-nums[i],value为index Map<Integer, Integer> map = new HashMap<>(); for (int i = 0; i < nums.length; i++) { map.put(target-nums[i],i); } int[] res = new int[2]; for (int i = 0; i < nums.length; i++) { Integer b; if ((b = map.get(nums[i])) != null && b != i) { res[0] = b; res[1] = i; break; } } return res; */ //Two //key为target-nums[i],value为index Map<Integer, Integer> map = new HashMap<>(); int[] res = new int[2]; for (int i = 0; i < nums.length; i++) { if (map.containsKey(nums[i])) { res[0] = map.get(nums[i]); res[1] = i; break; } map.put(target - nums[i], i); } return res; } public static void main(String[] args) { int[] nums = new int[]{2, 7, 11, 15}; System.out.println(Arrays.toString(TwoSum.twoSum(nums, 9))); } }