给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。你可以按任意顺序返回答案。
首先该题要求时间复杂度低于O(n^2),可以采用空间换时间的方法采用哈希表的方式。
key为target与num[]中值之差,value为num[]中值的下标。检查哈希表中是否存在target与当前值之差:如果存在则返回该结果,如果不存在则将其存入哈希表中。
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<>();
for(int i = 0; i < nums.length; i++){
if(map.containsKey(target - nums[i])) {
return new int[] {map.get(target - nums[i]), i};
}
map.put(nums[i], i);
}
//throw new IllegalArgumentException("No two sum solution");//题解中还有抛出异常的做法,但是不了解两者优劣,希望有大神不吝赐教
return new int[0];
}
}
使用哈希表的方法很简单易懂,需要注意的就是哈希冲突的问题,上述方法可以有效避开hash冲突。每次写入时,判断条件不是当前的key本身存不存在,而是key和tag之间的差值存不存在,这一点很重要。假定只有一个解,也就是说重复元素再多都无所谓。如果有3个或者以上的重复元素,代表这个重复元素不可能是解,所以写入map的时候直接覆盖也无所谓;如果只有两个重复元素,同样的道理,假如这个重复元素是解,那么必定是两个重复元素的和等于tag。这种情况下,当遇到第二个重复元素时,不会写入map,这个时候直接已经取到解了。 所以这种方式,不用考虑hash冲突的问题。