1171. 從鏈表中刪去總和值為零的連續(xù)結點
給你一個鏈表的頭結點 head,請你編寫代碼,反復刪去鏈表中由 總和 值為 0 的連續(xù)結點組成的序列,直到不存在這樣的序列為止。
刪除完畢后,請你返回最終結果鏈表的頭結點。
示例:
輸入:head = [1,2,-3,3,1]
輸出:[3,1]
提示:答案 [1,2,1] 也是正確的。
解題思路:
找到題目的關鍵點:和值為0的連續(xù)結點。我們來看看示例,如果我們選擇[-3, 3]這個區(qū)間,那么從左至右累加各個結點的值在這個區(qū)間前后是沒有變化的。也就是說,用list來表示鏈表的累加和:[1, 3, 0, 3, 4],可以看到Node(2)和Node(3)處的累加和是相等的。這就是解題的關鍵,找到累加和不變的區(qū)間,讓區(qū)間前一個Node的下一個Node等于區(qū)間結束后緊跟的下一個Node。
代碼如下:
class Solution:
def removeZeroSumSublists(self, head: ListNode) -> ListNode:
dummy = ListNode(0)
dummy.next = head
sumMap = {}
node = dummy
sum_node = 0
while node:
sum_node += node.val
sumMap[sum_node] = node
node = node.next
sum_node = 0
node = dummy
while node:
sum_node += node.val
node.next = sumMap[sum_node].next
node = node.next
return dummy.next
時間復雜度:O(N),遍歷結點;
空間復雜度:O(N) ,存儲結點的累加和。
解法轉自 Java HashMap 兩次遍歷即可