两数之和
平均星级:4.68 (272次评分)
2018年5月27日 | 34.4K次 预览
给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
我的想法,直接爆破,两个for。
class Solution { public int[] twoSum(int[] nums, int target) { for(int i=0; i<nums.length; i++) for(int j=i+1; j<nums.length; j++) { if(nums[i] + nums[j] == target) { return new int[] { i, j }; } } throw new IllegalArgumentException("No two sum solution"); }}
这里值得注意的是,我忘了返回一个数组怎么写了,一开始我是申请了一个新的数组,然后放结果返回,当然很慢;
然后我看了看答案,知道这么返回了,OK;但是这是循环里的返回,我在外面也需要返回值,官方答案处理很聪明,直接抛出异常。
接着我看了答案,第二种,一层循环建立哈希表,一层循环找,这里我也想到了第三种,直接建立哈希表的时候找,官方还有一点我想不到的是nums[j] == target - nums[i]
最终代码:
class Solution { public int[] twoSum(int[] nums, int target) { Map<Integer, Integer> total = new HashMap<>(); for(int i=0; i<nums.length; i++) { int to = target - nums[i]; if(total.containsKey(to)) { return new int[]{total.get(to), i}; } total.put(nums[i], i); } throw new IllegalArgumentException("No two sum solution"); }}
有点气,我之前的判断不对,没想到containsKey这个函数。。。。
我的判断是total.get(to) != null && total.get(to) < total.size()这个不能通过,原因我要再想想