leetcode 1——兩數之和

問題描述

給定一個整數數組 nums 和一個目標值 target，請你在該數組中找出和爲目標值的那 兩個 整數，並返回他們的數組下標。

你能夠假設每種輸入只會對應一個答案。可是，你不能重複利用這個數組中一樣的元素。

示例:

給定 nums = [2, 7, 11, 15], target = 9 由於 nums[0] + nums[1] = 2 + 7 = 9 因此返回 [0, 1]

嘗試解法

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums) - 1):
            for j in range(i+1, len(nums)):
                if nums[i] + nums[j] == target:
                    return i, j

#暴力解決，依次遍歷每一個元素，找到和爲target的數，返回其下標

官方題解

兩遍哈希表
爲了對運行時間複雜度進行優化，咱們須要一種更有效的方法來檢查數組中是否存在目標元素。若是存在，咱們須要找出它的索引。保持數組中的每一個元素與其索引相互對應的最好方法是什麼？哈希表。

經過以空間換取速度的方式，咱們能夠將查找時間從 O(n)O(n) 下降到 O(1)O(1)。哈希表正是爲此目的而構建的，它支持以 近似 恆定的時間進行快速查找。我用「近似」來描述，是由於一旦出現衝突，查找用時可能會退化到 O(n)O(n)。但只要你仔細地挑選哈希函數，在哈希表中進行查找的用時應當被攤銷爲 O(1)O(1)。

一個簡單的實現使用了兩次迭代。在第一次迭代中，咱們將每一個元素的值和它的索引添加到表中。而後，在第二次迭代中，咱們將檢查每一個元素所對應的目標元素（target - nums[i]target−nums[i]）是否存在於表中。注意，該目標元素不能是 nums[i]nums[i] 自己！

　Java版　

public int[] twoSum(int[] nums, int target) {
    Map<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        map.put(nums[i], i);
    }
    for (int i = 0; i < nums.length; i++) {
        int complement = target - nums[i];
        if (map.containsKey(complement) && map.get(complement) != i) {
            return new int[] { i, map.get(complement) };
        }
    }
    throw new IllegalArgumentException("No two sum solution");
}