1.两数之和

题目说明

难度简单

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

解题思路

map[target - nums[idx]] = idx 代表map中key 放入的是 target - nums所在index的value值，而key中的value 为 这个被减数的index。

举例：

target = 9

map[9 - nums[0]] = 0   //→→→   map[7] = 0
        ⬇️
nums[1] = 7
map[7] = 0 //有值！！

return [0, 1]

func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
    var map: [Int: Int] = [:]
    for idx in 0..<nums.count {
        map[target - nums[idx]] = idx
    }
    for idx in 0..<nums.count {
        if let value = map[nums[idx]], idx != value {
            return [idx, value]
        }
    }
    return []
}

因此代码可以优化如下：

func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
    var map: [Int: Int] = [:]
    for idx in 0..<nums.count {
        let subtraction = target - nums[idx]
        let value = nums[idx]
        if let lidx = map[value] {
            return [lidx, idx]
        }
        map[subtraction] = idx
    }
    return []
}

代码优化一下

func twoSum(_ nums: [Int], _ target: Int) -> [Int] {
    // 1. 初始化哈希表
    var hash = [Int: Int]()
    
    // 2. 遍历数组
    for (i, num) in nums.enumerated() {
        // 3. 计算目标补数
        let complement = target - num
        
        // 4. 检查补数是否存在
        if let complementIndex = hash[complement] {
            // 5. 返回索引
            return [complementIndex, i]
        }
        
        // 6. 存储当前元素索引
        hash[num] = i
    }
    
    // 7. 处理无匹配情况
    return []
}