题目大意:找nums数组中的三个数,使它们的和最接近目标值target,并且只有一组解。
题目分析:首先想到的是暴力解法,遍历出所有从数组中取不同的三个数的情况,比较它们与target的距离(可以用绝对值表示),然后将距离最小的一组的和输出即可。这种方法是超时的,简单分析一下,可以知道时间复杂度为O(n^3).
通过分析,我们可以想到一种时间复杂度为O(n^2)的解法:假设数组中有len个元素,首先我们将数组中的元素按照从小到大的顺序进行排序。其次,看最终取出的三个数中的第一个数,若数组长度为n,那么有n种取法。假设取的第一个数是A[i],那么第二三两个数从A[i+1]~A[len]中取出。找到“第一个数为A[i]固定,后两个数在A[i]后面元素中取。并且三数之和离target最近的情况。”这时,我们用两个指针j,k分别指向A[i+1]和A[len],如果此时三数之和A[i]+A[j]+A[k]<target,说明三数之和小了,我们将j后移一格;反之,若和大于target,则将k前移一格;直到j和k相遇为止。在这期间,保留与target最近的三数之和。一旦发现有“和等于target的情况”,立即输出即可。
由于取的第一个数可以是A[0],A[1],A[2],……,A[len-1],所以需要重复以上步骤n次。
为什么第一个数取了A[i]之后,第二三两个数只能在A[i+1]~A[len]中取呢? 因为这样可以避免重复。假设第二个数取了A[i-2],那么这样情况势必会包含在第一个数取A[i-2]的情况中。因为取出的三个数之间是没有顺序关系的。
代码展示:
class Solution:
def threeSumClosest(self, nums, target):
result = nums[0]+nums[1]+nums[2]
dis = abs(result-target)
nums.sort()
for i in range(len(nums)):
num = nums[i]
j = i+1
k = len(nums)-1
while j<k:
temp = abs(nums[i]+nums[j]+nums[k]-target)
if temp<dis:
dis = temp
result = nums[i]+nums[j]+nums[k]
if nums[i]+nums[j]+nums[k]<target:
j += 1
elif nums[i]+nums[j]+nums[k]>target:
k -= 1
else:
return nums[i]+nums[j]+nums[k]
return result