两数之和

2019-03-18  本文已影响0人  Jabir_Zhang

题目:两数之和

描述

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

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

示例:

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

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

方法一:暴力法

暴力法很简单。遍历每个元素 xx,并查找是否存在一个值与 target - x相等的目标元素。

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

方法二:两遍哈希表

使用字典,时间复杂度O(n),空间复杂度O(n)

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 []
    }

方法三:一遍哈希表

方法二的优化

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 []
    }
上一篇下一篇

猜你喜欢

热点阅读