手撕LeetCode #1171——Python

1171. 從鏈表中刪去總和值為零的連續(xù)結(jié)點(diǎn)
給你一個(gè)鏈表的頭結(jié)點(diǎn) head,請(qǐng)你編寫(xiě)代碼,反復(fù)刪去鏈表中由 總和 值為 0 的連續(xù)結(jié)點(diǎn)組成的序列,直到不存在這樣的序列為止。
刪除完畢后,請(qǐng)你返回最終結(jié)果鏈表的頭結(jié)點(diǎn)。

示例:

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

解題思路:
找到題目的關(guān)鍵點(diǎn):和值為0的連續(xù)結(jié)點(diǎn)。我們來(lái)看看示例,如果我們選擇[-3, 3]這個(gè)區(qū)間,那么從左至右累加各個(gè)結(jié)點(diǎn)的值在這個(gè)區(qū)間前后是沒(méi)有變化的。也就是說(shuō),用list來(lái)表示鏈表的累加和:[1, 3, 0, 3, 4],可以看到Node(2)和Node(3)處的累加和是相等的。這就是解題的關(guān)鍵,找到累加和不變的區(qū)間,讓區(qū)間前一個(gè)Node的下一個(gè)Node等于區(qū)間結(jié)束后緊跟的下一個(gè)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

時(shí)間復(fù)雜度:O(N),遍歷結(jié)點(diǎn);
空間復(fù)雜度:O(N) ,存儲(chǔ)結(jié)點(diǎn)的累加和。

解法轉(zhuǎn)自 Java HashMap 兩次遍歷即可

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

友情鏈接更多精彩內(nèi)容