从今天开始努力学Java
刚看到题目还想着,这不哈希表嘛,感觉自己也太有实力了,分分钟写下如下代码:
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer, Integer> hashmap= new HashMap<Integer, Integer>();
int[] arr=new int[2];
for(int i=0;i<nums.length;i++)
{
hashmap.put(nums[i],i);
if(hashmap.containsKey(target - nums[i]))
{
arr[1]=hashmap.get(target - nums[i]);
arr[0]=i;
}
}
return arr;
}
}
然后运行发现,如果输入为[3,3],target=6,那么这代码将会返回[1,1]而不是[0,1],发现关键在于应hashmap.put的时机,如果放在判断语句后面的话就是正确的。原因应该是如果put放在前,i=0的时候将3已经hash到hashmap中,判断语句target-3也就已经存在哈希表中,并且循环继续所以结果输出了[1,1],所以应该先判断再put。
for(int i=0;i<nums.length;i++)
{
if(hashmap.containsKey(target - nums[i]))
{
arr[1]=hashmap.get(target - nums[i]);
arr[0]=i;
}
hashmap.put(nums[i],i);
}