为什么我的方法会在列表中创建一个额外的列表？ Leetcode 3Sum 问题

Question

这是leetcode 3sum问题的解决方案：https://leetcode.com/problems/3sum/

我想我有一个解决方案，可能会很慢，但仍然是一个解决方案。但我似乎无法弄清楚为什么我不断收到额外的清单。基本上为input:nums = [-1,0,1,2,-1,-4]
我应该从我的代码中返回输出：[[-1,-1,2],[-1,0,1]].
但我得到的是：[[-1, 2, 1], [-1, 1, 0], [-1, 1, 0]]。

我尝试使用它来删除重复的列表，但我认为有一些逻辑错误导致了这个额外的列表。这是删除列表中重复列表的代码，没有提供任何有用的输出，只是内存位置。

代码：

dup_set = set(map(tuple,a))
non_dup_set = map(list,dup_set)

我还想我可能需要一个 hashmap，以便该方法知道不重用索引，但我对 hashmap 的理解不是很好，我只是认为键值，但除此之外没有。

这是完整的代码：

class Solution:
    def threeSum(self, nums):
        nums.sort()
        l = 0
        r = len(nums) - 1
        final_list = []
        
        while l < r:
            if nums[l] + nums[r] <= 0:
                print(nums[l+1:r])
                if (nums[l] + nums[r]) in nums[l+1:r]:
                    final_list.append([nums[l],nums[r],(nums[l]+nums[r])])
                l+=1
            elif nums[l] + nums[r] >= 0:
                print(nums[l+1:r])
                if -(nums[l] + nums[r]) in nums[l+1:r]:
                    final_list.append([nums[l],nums[r],(nums[l]+nums[r])])
                r-=1

        print(final_list)
            
            
Solution().threeSum([-1,0,1,2,-1,-4])

输出：[[-1, 2, 1], [-1, 1, 0], [-1, 1, 0]]

所需：[[-1, 2, 1], [-1, 1, 0]]

请帮忙 谢谢

Answer 1

我认为你可以这样：

non_dup_set = list(set(map(tuple, a)))

Answer 2

考虑这个答案。它得到正确的元组，尽管顺序不同。不知道为什么这很重要。

class Solution:
    def threeSum(self, nums):
        final_list = []
        for l, v1 in enumerate(nums):
            for r, v2 in enumerate(nums[l+1:]):
                if -(v1+v2) in nums[l+r+1:]:
                    check = sorted((v1,v2,-(v1+v2)))
                    if check not in final_list:
                        final_list.append(check)
        print(final_list)

Solution().threeSum([-1,0,1,2,-1,-4])

更好的答案是使用itertools.combinations 来简化事情：

import itertools
class Solution:
    def threeSum(self, nums):
        final_list = []
        for subset in itertools.combinations(nums,3):
            if sum(subset) == 0:
                check = sorted(subset)
                if check not in final_list:
                    final_list.append(check)
        print(final_list)

Solution().threeSum([-1,0,1,2,-1,-4])

以及强制性的单行：

import itertools
class Solution:
    def threeSum(self, nums):
        final_list = list(set(tuple(sorted(subset)) for subset in itertools.combinations(nums,3) if sum(subset) == 0))
        print(final_list)

Solution().threeSum([-1,0,1,2,-1,-4])