手撕LeetCode #1171——Python

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 兩次遍歷即可

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

友情鏈接更多精彩內容