算法每日一刷

LeetCode算法题-16. 最接近的三数之和(Swift)

2019-09-29  本文已影响0人  entre_los_dos

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum-closest
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.

与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

方法一:三个for循环。太耗时

func threeSumClosest(_ nums: [Int], _ target: Int) -> Int {
        if nums.count < 3 {
            return 0
        }
        
        var maxSum = nums[0] + nums[1] + nums[2]
        
        for i in 0..<nums.count - 2 {
            for j in i+1..<nums.count - 1 {
                for k in j+1..<nums.count {
                    
                    let currentSum = nums[i] + nums[j] + nums[k]
                    
                    if abs(maxSum - target) > abs(currentSum - target) {
                        maxSum = currentSum
                    }
                }
            }
        }
        return maxSum
    }

方法二:一个for循环,里面的两个for循环变成两个指针找

func threeSumClosest(_ nums: [Int], _ target: Int) -> Int {
        if nums.count < 3 {
            return 0
        }
        
        let sortNums = nums.sorted()
                
        var resultSum = sortNums[0] + sortNums[1] + sortNums[2]

        for i in 0..<sortNums.count - 2 {
            
            var startIndex = i+1
            var endIndex = sortNums.count - 1
            while startIndex < endIndex {
                
                let currentSum = sortNums[i] + sortNums[startIndex] + sortNums[endIndex]
                
                if currentSum > target {
                    endIndex -= 1
                }else {
                    startIndex += 1
                }
                if abs(resultSum - target) > abs(currentSum - target) {
                    resultSum = currentSum
                }
                
                if abs(resultSum - target) == 0 {
                    return resultSum
                }
            }
        
        }
        return resultSum
    }
上一篇下一篇

猜你喜欢

热点阅读