LeetCode(swift) - 两数之和

题目描述

-给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
-你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例：

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

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

暴力法很简单，遍历每个 num，查找出与 target - num 相等的另外一个 num， 时间复杂度为O(n^2)

利用 Dictionary 进行求解，因为hashmap 查找时间为 O(1)，将匹配的速度大大加快了，摘抄网上的一些方案如下

class Solution {
    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], i != j {
                return [i, j]
            }
        }
        return []
    }
}

注意注意！

上述方案是有问题的，不知道跑过case的同学有没有注意到，Dictionary 的 key 必须是惟一的，而数组中的元素则不一定，所以上述方案去跑 nums = [3, 3], target = 6 这个case的话，是通不过的

个人方案：

class Solution {
    func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
        var numMap: [Int: Int] = [:]
        for i in 0..<nums.count {
            let matchNum = target - nums[i]
            if let matchIdx = numMap[matchNum] {
                return [matchIdx, i]
            } else {
                numMap[nums[i]] = i
            }
        }
        return []
    }
}

规避了key重复的问题，另外不需要把整个数组进行map，在空间复杂度上面有所优化，如果有更好的方案，欢迎留言交流。