[LeetCode]Two Sum

<span style="font-family:SimSun;">如雷贯耳的Two Sum,光听隔壁哥们店面别人就面了两遍,真到自己写的时候还是没能一遍通过,事非经过不知难!算法思想很简单,用一个HashMap把数字出现的位置记下来,但自己第一遍提交时忘了考虑同一个数等于target的一半的情况,比如target是6,numbers是{3, 2, 4},我返回了1,1, 这是不对的。 Accepted 答案如下,红色部分为忘了check的条件</span>

import java.util.Map;
import java.util.HashMap;

public class Solution {
  public int[] twoSum(int[] numbers, int target) {
    Map<Integer, Integer> hash = new HashMap<Integer, Integer>();
    for (int i = 0; i < numbers.length; i++){
      hash.put(target - numbers[i], i + 1);
    }
    for (int i = 0; i < numbers.length; i++){
      if (hash.get(numbers[i]) != null && hash.get(numbers[i]) != i + 1){       
        return new int[]{i +1, hash.get(numbers[i])};
      }
    }
    return new int[]{-1, -1};
  }
}

但是这个答案循环了两遍,其实官方给的答案可以一个循环搞定:

<pre name="code" class="java">public static int[] twoSum(int[] numbers, int target){
  Map<Integer, Integer> hash = new HashMap<Integer, Integer>();
  for (int i = 0; i < numbers.length; i++){
    int x = numbers[i];
    if (hash.containsKey(target - x)){
      return new int[]{hash.get(target -x) + 1, i + 1};
    }
    hash.put(x, i);
  }
  return new int[]{-1, -1};
}


不管怎样,第一题正式开始了!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章