【20190707】【每天一道算法题】最接近的三数之和（双指针）

问题

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

例如，给定数组 nums = [-1，2，1，-4]，和 target = 1。与 target 最接近的三个数的和为 2。即(-1 + 2 + 1 = 2)。

---

思路及解答

1. Python实现

# 方法一：暴力法（三次遍历）
# 时间复杂度：O(n^3)
# 超时，略


# 方法二：双指针法（两次遍历）
# 在每次循环中，首先固定第一个指针 p，然后定义双指针 i, j 分别指向 p+1 和 end。然后计算三者之和和 target 比较。若和小于 target，则i++，否则j--。
# 时间复杂度：O(n^2)
class Solution:
    def threeSumClosest(self, nums: List[int], target: int) -> int:
        l = len(nums)
        nums.sort()
        result = float("inf")
        i, j= 1, l-1
        for p in range(l-2):   # p 是 [0, l-2]，i 是 [p+1, j]，j 是 [i, end]
            i, j = p+1, l-1
            while i < j:
                tmp = nums[p] + nums[i] + nums[j]
                if tmp == target:
                    return tmp
                elif tmp < target:
                    result = tmp if abs(tmp-target) < abs(result-target) else result
                    i = i + 1
                else:
                    result = tmp if abs(tmp-target) < abs(result-target) else result
                    j = j - 1
        return result

---

知识点

1. 列表没有求绝对值的这种操作：abs(list)，但可以用 map 函数实现：absList = map(abs, list)

参考：python中，怎样对列表中每一项求绝对值？

2. 同样地，让列表中每个元素减去同一个数，也不可以用 list-10 这种表达（但Matlab 可以，区别记忆！），只有用循环咯。