美文网首页
15.leetcode题目讲解(Python):三数之和

15.leetcode题目讲解(Python):三数之和

作者: 夏山闻汐 | 来源:发表于2018-09-04 23:09 被阅读290次

    题目:

    题目

    给定一个数列,求a+b+c = 0的问题可以转换为求:a+b = -c, 我们将输入的list排序后,给定nums[i], 我们需要寻找 nums[j]+nums[k] = - nums[i], 这里的 j,k > i

    题目要求不可包含重复的解,对于排序后的数组,如果num[i] = num[i+1],那么他们获得的nums[j]和nums[k]肯定是相同的(如果解存在的话),所以应该跳过这种情况。同理,如果num[j] = num[j+1], 那么也会找到相同的nums[k]。这种情况也应该跳过。

    参考代码如下:

    class Solution(object):
        def threeSum(self, nums):
            """
            :type nums: List[int]
            :rtype: List[List[int]]
            """
            nums.sort()
            L, res = len(nums), []
            for i in range(L-2):
                if i > 0 and nums[i] == nums[i-1]:
                    continue
                target = -1 * nums[i]
                j,k = i + 1, L - 1
                while j<k:
                    if nums[j]+nums[k] == target:
                        res.append([nums[i], nums[j], nums[k]])
                        j = j + 1
                        while j<k and nums[j] == nums[j-1]:
                            j = j + 1
                    elif nums[j] + nums[k] < target:
                        j = j + 1
                    else:
                        k = k - 1
            return res
    

    如果您有更好的解法,欢迎讨论。
    ps:如果您有好的建议,欢迎交流 :-D,也欢迎访问我的个人博客:tundrazone.com

    相关文章

      网友评论

          本文标题:15.leetcode题目讲解(Python):三数之和

          本文链接:https://www.haomeiwen.com/subject/ylfowftx.html