LeetCode刷題 - (01)兩數之和

題目描述

給定一個整數數組nums和一個目標值target，請你在該數組中找出和爲目標值的那兩個整數，並返回他們的數組下標。
你可以假設每種輸入只會對應一個答案。但是，你不能重複利用這個數組中同樣的元素。

示例:

給定 nums = [2, 7, 11, 15], target = 9
因爲 nums[0] + nums[1] = 2 + 7 = 9

所以返回 [0, 1]

解法一 暴力解法

解題思路

最容易想到的就是暴力法，使用兩個遍歷，查找兩數之和是否爲target。流程如下：

1. 使用i來遍歷數組每個元素
2. 在i的每一個循環中，使用j從i + 1開始遍歷
3. 判斷i和j對應的值的和是否爲target

代碼實現

func twoSum(nums []int, target int) []int {
    for i := 0; i < len(nums); i ++ {
        for j := i + 1; j < len(nums); j ++{
            if nums[i] + nums[j] == target{
                return []int{i,j}
            }
        }
    }
    return []int{}
}

複雜度分析

• 時間複雜度: O(n^2)。一共有兩次循環，每次循環的事件複雜度是O(n)。總的複雜度爲O(n^2)。
• 空間複雜度: O(1)。

解法二 哈希表解法

解題思路

該解法中會藉助一個哈希表。在遍歷數組時，會將當前元素添加到哈希表中。並且會檢查哈希表中是否已經存在當前元素所對應的目標元素。如果存在就返回。

代碼實現

func twoSum2(nums []int, target int) []int {
    m := make(map[int]int)
    for i := 0; i < len(nums); i ++ {
        if index, exists := m[target - nums[i]]; exists {
            return []int{i,index}
        }
        m[nums[i]] = i
    }
    return []int{}
}

複雜度分析

• 時間複雜度：O(n)。我們只遍歷了一次數組。
• 空間複雜度：O(n)。使用了哈希表，需要額外的空間。取決於哈希表中存儲的元素數量，該表最多需要存儲O(n)個元素。