1171. 从链表中删去总和值为零的连续节点
Difficulty
Medium
Tags
链表
Star
给你一个链表的头节点
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
.
通过次数15,701提交次数32,782
法1 hash
思路
用hash表格存储前缀和,key 为前缀和,val为该节点指针
所以hash中每个前缀和总是存储着靠后面的节点,因此即可删掉中间为0 的子链表
题解
# Definition for singly-linked list. # class ListNode: # def __init__(self, val=0, next=None): # self.val = val # self.next = next class Solution: def removeZeroSumSublists(self, head: ListNode) -> ListNode: dummy = ListNode(0, head) hash_map = {} cur, sum_ = dummy, 0 while cur: sum_ += cur.val hash_map[sum_] = cur cur = cur.next cur, sum_ = dummy, 0 while cur: sum_ += cur.val cur.next = hash_map[sum_].next cur = cur.next return dummy.next