16. 3Sum Closest

题目:

Given an array S of n integers, find three integers in S such that the sum is closest to a given number, target. Return the sum of the three integers. You may assume that each input would have exactly one solution.

    For example, given array S = {-1 2 1 -4}, and target = 1.

    The sum that is closest to the target is 2. (-1 + 2 + 1 = 2).

代码:

做过第15题:3Sum之后,看到该问题,不会很陌生,只不过这次目标值不在是0,而是尽可能接近一个传入的参数

同样先尝试了遍历3遍,也就是O(n^3)时间复杂度,不出所料,超时。

于是,根据上一题的经验,先确定第一个数字,之后在剩余数字中定义两端的两个指针,一个最大一个最小,根据三数相加和目标值比较的结果,分别移动左右两端的指针,增加或缩小三数的和,靠近目标值。

代码已经加入详细注释,根据之前的经验自己写的,调试了很久。还有很多不足,共同学习,望大神多多指教:

def threeSumClosest(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        if len(nums) < 3: return 0
        #排序
        nums.sort()
        res = 0
        result = []
        flag = []
        length = len(nums)
        for i in range(0, length - 2):
            if i and nums[i] == nums[i - 1]:
                continue
            left, right = i + 1, length - 1
            while left < right:   
                #如果三数字相加结果等于目标值,直接返回结果
                if nums[i] + nums[left] + nums[right] == target:
                    #print(left,right, "equal ",nums[i], nums[left],  nums[right])
                    res = target
                    return res
                #如果三数字相加结果大于目标值,右边的数字指针减小
                elif nums[i] + nums[left] + nums[right] > target:
                    print(left,right, "big ",nums[i], nums[left],  nums[right])                      
                    res =  nums[i] + nums[left] + nums[right]
                    right -= 1
                    #如果指针指向的下一个数字和该数字相同,则指针继续向左移动
                    while left < right and nums[right]==nums[right+1]:
                        right -= 1
                    #由于存在移动一位之后三数字相加和突然小于目标值的情况,所以需要记录该值来比较
                    #否则下一轮循环之后的值和目标值得差可能比当前更大
                    if nums[i] + nums[left] + nums[right] < target:
                        result.append(res)
                        flag.append(abs(res-target))   
                #如果三数字相加结果小于目标值,左边的数字指针增加        
                else:
                    print("little ",nums[i], nums[left],  nums[right])                  
                    res =  nums[i] + nums[left] + nums[right]
                    left += 1
                    #如果指针指向的下一个数字和该数字相同,则指针继续向右移动
                    while left < right and nums[left]==nums[left-1]:
                        left += 1
                    #由于存在移动一位之后三数字相加和突然大于目标值的情况,所以需要记录该值来比较
                    #否则下一轮循环之后的值和目标值得差可能比当前更大
                    if nums[i] + nums[left] + nums[right] > target:
                        result.append(res)
                        flag.append(abs(res-target))   
            #result分别存储不同的第一个数字遍历后最接近目标值的结果
            #flag对应存储不同结果和目标值之间的差
            result.append(res)
            flag.append(abs(res-target))
        print (result,'\n',flag)
        #返回差距最小的结果即可    
        return (result[flag.index(min(flag))])

posted @ 2016-11-09 18:41  PolarBearInterest  阅读(121)  评论(0编辑  收藏  举报