弟中弟的Leetcode总结——数组类（三）

弟中弟的Leetcode总结——数组类（三）

题目描述

3Sum

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

The solution set must not contain duplicate triplets.

Example:

Given array nums = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

思路

这道题有一个naive solution，就是三重循环遍历整个数组，找出所有可能的值，同时再判断是否重复。直觉告诉我这样的做法一定会超时，所以没有尝试（然而好像是可以ac的哭）。
拿到这道题看了一下输出的结果，对于每一个list，输出都是有序的，因此想到了先将给的数组进行排序，然后再看看怎样进一步处理。
由于刚刚做完前面的一道求最大容量的题（请参看我的数组类（二）），想到了设置两个指针，一个指向前面，一个指向最后，然后遍历前(numSize-2)个数，然后根据这三个变量的值的和来进行判断，判断是移动前面的指针，还是后面的指针，还是遍历的变量。
（1）如果和为0，首先将这三个数直接加入answer的list，然后再判断指针1后面的数字和指针2前面的数字是否是重复的，如果是，那么直接跳过，然后再向前/向后移动一位
（2）如果和小于0，那么说明前面的指针需要向后移动
（3）同理如果和大于0，那么说明后面的指针需要向前移动

代码（py3）

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        leng=len(nums)
        ans=[]
        if(leng<3):
            return ans
        flag=0
        for i in range(leng-2):
            if(nums[i]>0):
                return ans
            if(i>0 and (nums[i]==nums[i-1])):
                i=i+1
                continue
            j=i+1
            k=leng-1

            while(j<k):
                res=nums[i]+nums[j]+nums[k]
                if(res==0):
                    ans.append([nums[i],nums[j],nums[k]])
                    while((nums[j]==nums[j+1]) and j<k):
                        j=j+1
                        if(j==k):
                            break
                    while((nums[k]==nums[k-1]) and j<k):
                        k=k-1
                    j=j+1
                    k=k-1
                elif(res<0):
                    j=j+1
                else:
                    k=k-1
        return ans