1171. 从链表中删去总和值为零的连续节点

最新推荐文章于 2023-07-16 19:19:12 发布

原创最新推荐文章于 2023-07-16 19:19:12 发布 · 177 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

110 篇文章

订阅专栏

这篇博客介绍了两种方法来删除链表中和为0的子列表。第一种方法通过构建前缀和数组，找到相等的前缀和并删除对应节点；第二种方法利用哈希表存储前缀和与节点的关系，直接更新链表。这两种方法都巧妙地解决了问题，有效地删除了满足条件的子列表。

暴力大法

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def removeZeroSumSublists(self, head: ListNode) -> ListNode:
        lst = []
        p = head
        while p:
            lst.append(p.val)
            p = p.next

        while True:
            i,j = 0,1
            flag = False
            while i<len(lst):
                if j<=len(lst):
                    if sum(lst[i:j]) == 0:
                        flag = True
                        del lst[i:j]
                    else:
                        j += 1
                else:
                    i += 1
                    j = i+1
            if flag == False:
                break

        nh = ListNode(-1)
        p = nh
        for i in range(len(lst)):
            p.next = ListNode(lst[i])
            p = p.next

        return nh.next

前缀和

方法比较巧妙
首先拼接一个假结点，然后建立前缀和<->结点的映射，然后对于重复的前缀和，覆盖最后一个节点
对于 1，2，3，-3，-2的数组可以得到前缀和
[0,1,3,6,3,1]
如果两个结点的前缀和相同，那么它们中间的结点和为0，可以直接删除
删除操作用next来更新即可

class Solution:
    def removeZeroSumSublists(self, head: ListNode) -> ListNode:
        dct = dict()
        dh = ListNode(0)

        dh.next = head
        p = dh
        sumv = 0
        while p:
            sumv += p.val 
            dct[sumv] = p
            p = p.next

        p = dh
        sumv = 0
        while p:
            sumv += p.val
            p.next = dct.get(sumv).next 
            p = p.next

        return dh.next