给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
输入:nums = [2,7,11,15], target = 9 输出:[0,1]
输入:nums = [3,2,4], target = 6 输出:[1,2]
输入:nums = [3,3], target = 6 输出:[0,1]
public static int[] twoSum(int[] nums, int target) { int[] index=new int[2]; Map<Integer,Integer> map=new HashMap<Integer,Integer>(); for(int i=0;i<nums.length;i++){ if(map.containsKey(nums[i])){ index[0]=(int)map.get(nums[i]); index[1]=i; } map.put(target-nums[i],i); } return index; }
改进:
将index数组写到if()方法体中,因为因为该数组在整个代码逻辑中只会用到一次,所以在if()方法体中进行数组的创建和初始化。但是在twoSum()方法的最后一行,添加return new int[0];返回一个数组引用。
第9行代码map.put(target-nums[i],i);应该改成map.put(nums[i],i);。因为前者将会将很多不属于nums[i]的数放进map中,属于无中生有,会额外增加内存。
public static int[] twoSum(int[] nums, int target) { //创建一个哈希表,存储键值对 Map<Integer,Integer> map=new HashMap<Integer,Integer>(); for(int i=0;i<nums.length;i++){ //如果哈希表中包含该key——target-nums[i],将target-nums[i]对应的value返回,再将i一起返回。 if(map.containsKey(target-nums[i])){ return new int[]{map.get(target-nums[i]),i}; } //如果哈希表中不包含该key——nums[i],将(数,数组下标)作为一个键值对添加到哈希表中。 map.put(nums[i],i); } return new int[0]; }
数组中同一个元素在答案里不能重复出现,是不是提示了将数组中的元素作为不能重复的key值,而数组下标作为value值。如果是这样,题目要求返回两个数组下标,即返回两个value值。进一步考虑到方法的返回类型就是数组类型int[];
最后一句return new int[0];就是返回一个int数组的引用,数组是空的,这样也可以减少空间。
map.containsKey方法:判断哈希表中是否存在该key值。
public boolean containsKey(Object key) { return getNode(hash(key), key) != null; }map.get方法:通过key值获得value值。
public V get(Object key) { Node<K,V> e; return (e = getNode(hash(key), key)) == null ? null : e.value; }map.put方法:将key值和value值成对的放入哈希表中。
public V put(K key, V value) { return putVal(hash(key), key, value, false, true); }