题目梗概:
在一个整型数组中找出两个元素,它们相加的和等于目标数,并返回由它们的下标组成的新的数组。
解题思路:
正常来说,这道题的思考方向都是先遍历数组,得到第一个相加数,然后用目标数减去第一个相加数,得到另一个相加数,再判断数组中是否有与这个相加数相同的数,有则返回由这两个相加数在数组中的下标组成的新的数组。解这道题的关键在于如何快速地判断数组中是否有另一个相加数?由于我采用的编程语言是Java,自然而然地,我会想到Java中有没有相应的数据结构能够支持这种操作?我想到了Java的集合类,它们都有contains()方法,能够快速地判断集合中是否有指定的元素存在,这正是我所需要的。我所需要做的就是先将数组中的元素存到集合中,然后调用集合的contains()方法判断集合中是否有另一个我所需要的数。
解题算法:
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> m = new HashMap<Integer, Integer>();
for(int i=0; i<nums.length; i++){
int sub = target - nums[i];
if(m.containsKey(sub)){
return new int[]{m.get(sub), i};
}
m.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
}