美文网首页
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