【Lintcode】56. Two Sum

題目地址：

https://www.lintcode.com/problem/two-sum/description

給定一個整數數組，求其中和爲給定數字target的數對的兩個下標。題目保證解唯一。

思路是用一個哈希表記錄每個數字出現的位置，然後邊遍歷邊查看target減去當前數字是否在哈希表裏。代碼如下：

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

public class Solution {
    /**
     * @param numbers: An array of Integer
     * @param target: target = numbers[index1] + numbers[index2]
     * @return: [index1, index2] (index1 < index2)
     */
    public int[] twoSum(int[] numbers, int target) {
        // write your code here
        Map<Integer, Integer> map = new HashMap();
        for (int i = 0; i < numbers.length; i++) {
            if (map.containsKey(target - numbers[i])) {
                return new int[]{map.get(target - numbers[i]), i};
            }
            // 記錄當前數字以及其出現的位置
            map.put(numbers[i], i);
        }
        
        return null;
    }
}

時空複雜度$O(n)$。