题解 | #牛牛的三元组问题# Python

可以使用两层循环来遍历数组中的两个数，然后再使用双指针法找到第三个数。具体的代码如下

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
#
# @param nums int整型一维数组
# @return int整型二维数组
#
class Solution:
    def findTriplets(self, nums: List[int]) -> List[List[int]]:
        nums.sort()  # 先将数组排序

        triplets = []
        n = len(nums)

        for i in range(n - 2):
            if i > 0 and nums[i] == nums[i - 1]:
                continue  # 避免重复的结果

            left = i + 1
            right = n - 1

            while left < right:
                total = nums[i] + nums[left] + nums[right]

                if total == 0:
                    triplets.append([nums[i], nums[left], nums[right]])

                    while left < right and nums[left] == nums[left + 1]:
                        left += 1
                    while left < right and nums[right] == nums[right - 1]:
                        right -= 1

                    left += 1
                    right -= 1
                elif total < 0:
                    left += 1
                else:
                    right -= 1

        return triplets

这样就能得到一个满足条件的三元组的列表，在每个三元组中，元素按照字典序排序。