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

140 阅读2分钟

给你一个链表的头节点 head,请你编写代码,反复删去链表中由 总和 值为 0 的连续节点组成的序列,直到不存在这样的序列为止。

删除完毕后,请你返回最终结果链表的头节点。

你可以返回任何满足题目要求的答案。

(注意,下面示例中的所有序列,都是对 ListNode 对象序列化的表示。)

示例 1:

输入:head = [1,2,-3,3,1]
输出:[3,1]
提示:答案 [1,2,1] 也是正确的。

示例 2:

输入:head = [1,2,3,-3,4]
输出:[1,2,4]

示例 3:

输入:head = [1,2,3,-3,-2]
输出:[1]

提示:

  • 给你的链表中可能有 1 到 1000 个节点。
  • 对于链表中的每个节点,节点的值:-1000 <= node.val <= 1000.

核心代码

class Solution(object):
    def removeZeroSumSublists(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        values = []
        while head:
            values.append(head.val)
            head = head.next
        
        for k in range(len(values),0,-1):
            while True:
                for i in range(len(values) - k + 1):
                    if sum(values[i:i + k]) == 0:
                        values = values[:i] + values[i + k:]
                        break
                else:
                    break
        head = node = ListNode(0)
        for v in values:
            node.next = ListNode(v)
            node = node.next
        return head.next

第二种解法

class Solution(object):
    def removeZeroSumSublists(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        ans = ListNode(0)
        ans.next = head
        while True:
            hashmap = {0:ans}
            sum_ = 0
            node = ans.next
            while node:
                sum_ += node.val
                if sum_ not in hashmap:
                    hashmap[sum_] = node
                else:
                    hashmap[sum_].next = node.next
                    break
                node = node.next
            else:
                break
        return ans.next

第三种解法

class Solution(object):
    def removeZeroSumSublists(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        ans = ListNode(0)
        ans.next = head
        hashmap = {0:ans}
        sum_ = 0
        
        while head:
            sum_ += head.val
            hashmap[sum_] = head
            head = head.next
        
        head = ans
        sum_ = 0

        while head:
            sum_ += head.val
            head.next = hashmap[sum_].next
            head = head.next
        return ans.next

image.png

解题思路:第一种解法:我们先将链表转化成数组之后,我们在数组中完成到k个连续节点的和为0,我们就将其从数组中删除,然后剩下的节点,我们将其在构造成新的链表,暴力循环求解,效率比较低,可通过;第二种解法:用一个变量sum_记录前缀和,再用一个哈希表记录出现过的前缀和,如果出现了两个相同的前缀和,就说明中间这一段的和为0,是多余的.举例:对于输入 [1, 2, -2, 3, -1, -1, -1],前缀和为[1, 3, 1, 4, 3, 2, 1],下标为0的1和下标为2的1相同,就代表下标在【1, 2】间的元素的和为0。第三种解法:使用hashmap完成两次遍历:哈希表两次遍历,第一次遍历得到所有的值和节点的关系,第二次我们和第二种解法一样,我们取两个数字相同的节点,节点相同,中间之间next挂载,然后最终返回一个新链表。