LeetCode刷题笔记：字节跳动-数组和排序(1)-三数之和

欢迎大家光临我的个人博客http://www.chenjingjiu.cn，大家一起交流学习

1.三数之和

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

        例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

      

本题最直观的思路就是三个循环，确定前两个的和，然后找第三个数。但是这样的算法复杂度达到了O(n^3),需要对其进行优化，否则超出时间限制。于是想到排序原数组，然后设置两个指针l,r分别指向数组开头和结尾，第三个指针m指向l+1的位置。当前3个数的和tmp为l+r+m指针所指的值。如果tmp<0,m++,如果tmp>0,r--。中间还需要对于相同元素进行跳过处理，代码如下：

        class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        res = []
        nums.sort()
        l = len(nums)
        for i in xrange(l):
            if nums[i]>0:
                break
            if i>=1 and nums[i]==nums[i-1]:
                continue
            j,k= i+1,l-1
            while j0:
                    k-=1
                else:
                    j+=1
        return res