1. 程式人生 > >LeetCode 倆數之和 - Swift版本程式碼

LeetCode 倆數之和 - Swift版本程式碼

給定一個整數陣列和一個目標值,找出陣列中和為目標值的兩個數。

你可以假設每個輸入只對應一種答案,且同樣的元素不能被重複利用。

示例:

給定 nums = [2, 7, 11, 15], target = 9

因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

第一種:
暴力破解
* 遍歷每個元素 x,並查詢是否存在一個值與 target − x 相等的目標元素。

func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
     for i in 0..<nums.count {
         for
j in (i + 1)..<nums.count { if nums[j] == target - nums[i] { return [i,j] } } } return [] }

複雜度分析:

  • 時間複雜度:O(n​2), 對於每個元素,我們試圖通過遍歷陣列的其餘部分來尋找它所對應的目標元素,這將耗費 O(n) 的時間。因此時間複雜度為 O(n​2)。

  • 空間複雜度: O(1)。


第二種寫法:
為了對執行時間複雜度進行優化,我們需要一種更有效的方法來檢查陣列中是否存在目標元素。如果存在,我們需要找出它的索引。保持陣列中的每個元素與其索引相互對應的最好方法是什麼?雜湊表。

一個簡單的實現使用了兩次迭代。在第一次迭代中,我們將每個元素的值和它的索引新增到表中。然後,在第二次迭代中,我們將檢查每個元素所對應的目標元素(target−nums[i])是否存在於表中。注意,該目標元素不能是 nums[i] 本身!

func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
  let count = nums.count
    var dic = [Int: Int]()

    // 先把所有的數字 裝進 字典中
    for i  in 0..<count {
        dic[nums[i]] = i
    }

    for
i in 0..<count { let found = target - nums[i] if let j = dic[found], j != i { return [i,j] } } return[] }

複雜度分析:

  • 時間複雜度:O(n), 我們把包含有 n 個元素的列表遍歷兩次。由於雜湊表將查詢時間縮短到 O(1) ,所以時間複雜度為 O(n)。

  • 空間複雜度:O(n), 所需的額外空間取決於雜湊表中儲存的元素數量,該表中儲存了 n 個元素。


第三種:一遍 雜湊表
* 在進行迭代並將元素插入到表中的同時,我們還會回過頭來檢查表中是否已經存在當前元素所對應的目標元素。如果它存在,那我們已經找到了對應解,並立即將其返回。

func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
  let count = nums.count
    var dic = [Int: Int]()

    for i in 0..<count {
        let found = target - nums[i]
        if let j = dic[found],
            j != i {
            return [i,j]
        }
        dic[nums[i]] = i
    }

    return[]
}

複雜度分析:

  • 時間複雜度:O(n), 我們只遍歷了包含有 n 個元素的列表一次。在表中進行的每次查詢只花費 O(1) 的時間。

  • 空間複雜度:O(n), 所需的額外空間取決於雜湊表中儲存的元素數量,該表最多需要儲存 n 個元素。