leetcode 001 两数之和

2020-04-11  本文已影响0人  搜捕儿

两数之和

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

你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

示例:

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

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

解答:

第一次提交: 超时了...


681586607307_.pic_hd.jpg

正确解答:

// method 1 : 执行用时56ms
func twoSum(_ nums : [Int] , _ target : Int) -> [Int] {
    
    //var dict = Dictionary<Int,Int>()
    var dic = [Int:Int]()
    var i = 0
    for n in nums { //遍历数组将每一项数组值作为key,对应的数组下标作为value添加到字典里
        dic[n] = i
        i += 1
    }
    
    var result = [-1, -1]
    var j = 0
    for n in nums { //遍历数组,直接在上述字典中依次查找key为(target - n)的value,找到此value且与j值不同,即为答案
        if dic.keys.contains(target - n) && j != dic[target - n] {
            result[0] = j
            result[1] = dic[target - n] ?? -1
        }
        j = j + 1
    }
    
    return result
}

let arr = [2, 7, 11, 8]
let result_arr = twoSum(arr, 18)
print(result_arr)

// method 2 : 执行用时616ms
func twoSum2(nums : [Int], target : Int) -> [Int] {
    
    for i in 0..<nums.count-1 {
        let nn = nums[i]
        for j in i+1..<nums.count {
            let mm = nums[j]
            if nn + mm == target
            {
                return [i,j]
            }
        }
    }
    return [-1,-1]
}

let arr2 = [2, 7, 11, 8]
let result_arr2 = twoSum2(nums: arr2, target: 15)
print(result_arr2)

// method 3 : 耗时同方法一,但是消耗内存更小一些
func twoSum3 (nums : [Int], target : Int) -> [Int] {
    
    var dict : [Int : Int] = [:]
    for (index, value) in nums.enumerated() {
        if let dValue = dict[target - value] {
            return [dValue, index]
        }
        dict[value] = index
    }
    
    return []
    
}

let arr3 = [2, 7, 11, 8]
let result_arr3 = twoSum3(nums: arr2, target: 15)
print(result_arr3)

关于for-in

Swift取消了OC中的C形式的for循环。只用一种单一的for - in形式来取代

其中关于下标的获取方式

方式一: 通过enumerate来获取

let testArr = [9,7,3,8,5,2,1,0,6]
for num in testArr {
    print(num)
}

for num in testArr.enumerated() {
    print(num.element,num.offset)
}

方式二: 使用元组来方便获取元素索引

for (n, c) in "Swift".characters.enumerated() {
           print("\(n): '\(c)'")
       }
    ///     // Prints "0: 'S'"
    ///     // Prints "1: 'w'"
    ///     // Prints "2: 'i'"
    ///     // Prints "3: 'f'"
    ///     // Prints "4: 't'"
上一篇 下一篇

猜你喜欢

热点阅读