标签:哈希 2018年 shm ota 循环 原因 turn null i+1
平均星级: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()这个不能通过,原因我要再想想
标签:哈希 2018年 shm ota 循环 原因 turn null i+1
原文地址:http://blog.51cto.com/8641689/2147088