Study Notes
  • Kuma Blog
  • AI
    • AI-Resources
    • AI-books
    • Prompts
      • Prompts Free Courses
  • Movies
    • 2024
    • 2024
    • 2024
  • Google
    • chunked-palindrome
  • Setup
    • How to add a new user into Ubuntu and setup ssh key?
    • How to set up VSCode remote server connect with browser with Docker
  • kubernetes
  • Books
    • Designing-Data-Intensive-Applications
      • 第一章 — 可靠性,可扩展性,可维护性的应用程序(Reliable, Scalable, and Maintainable Applications)
    • System-Performance
      • Design-Data-Intensive-Application
      • Chapter 2: Methodologies
  • Languages
    • japanese
      • japanese-week
  • Leetcode
    • 30DayChallenge
      • LRU-cache
      • backspace-string-compare
      • binary-tree-maximum-path-sum
      • bitwise-and-number-range
      • check-string-valid-sequence-from-root-to-leaves-path-in-bst
      • construct-binary-search-tree-from-preorder-traversal
      • contiguous-array
      • counting-elements
      • diameter-of-binary-tree
      • first-unique-number
      • group-anagrams
      • jump-game
      • last-stone-weight
      • leftmost-column-with-at-least-a-one
      • longest-common-subsequect
      • maximal-square
      • maximum-subarray
      • middle-of-the-linked-list
      • min-stack
      • minimun-path-sum
      • move-zeroes
      • perform-string-shifts
      • product-of-array-except-itself
      • search-in-rotated-sorted-array
      • subarray-sum-equals-k
      • valid-parenthesis-string
    • English Solution
      • 1168.optimize-water-distribution-in-a-village-en
      • 1171.remove-zero-sum-consecutive-nodes-from-linked-list-en
      • 1177.can-make-palindrome-from-substring-en
      • 1343.number-of-avg-subarr-sizek-greater-or-equal-threshold
      • 1345.jump-game-iv
      • 25.reverse-nodes-in-k-groups-en
      • 474.ones-and-zeros-en
      • 53.maximum-sum-subarray-en
      • 547.friend-circles-en
      • 79.word-search-en
    • May2020Challenge
      • check-if-straight-line
      • cousins-in-binary-tree
      • find-town-judge
      • first-bad-version
      • first-unique-character-in-a-string
      • flood-fill
      • implement-trie
      • jewels-and-stones
      • majority-element
      • maximum-sum-circular-subarray
      • number-complement
      • odd-even-linkedlist
      • ransom-note
      • remove-k-digits
      • single-element-in-sorted-array
      • valid-perfect-square
    • python
      • 000017-Letter-Combinations-of-a-Phone-Number
      • 000032-Longest-Valid-Parentheses
      • 000033-Search-in-Rotated-Sorted-Array
      • 000046-Permutations
      • 000074-Search-a-2D-Matrix
      • 000077-Combinations
      • 000081-Search-in-Rotated-Sorted-Array-II
      • 000137-single-number-ii
      • 000139-Word-Break
      • 000207-courses-schedule
      • 000209-Minimum-Size-Subarray-Sum
      • 000376-wiggle-subsequence
      • 000445-Add-Two-Numbers-II
      • 000486-Predict-the-Winner
      • 000518-Coin-Change-II
      • 000673-Number-of-Longest-Increasing-Subsequence
      • 000688-Knight-Probability-in-Chessboard
      • 000735-Asteroid-Collision
      • 000852-Peak-Index-in-a-Mountain-Array
      • 859-Buddy-Strings
      • 000864-Shortest-Path-to-Get-All-Keys
      • 000920-Number-of-Music-Playlists
      • 001218-Longest-Arithmetic-Subsequence-of-Given-Difference
      • 001235-Maximum-Profit-in-Job-Scheduling
      • 001493-Longest-Subarray-of 1-After-Deleting-One-Element
      • Problem
      • 002024-Maximize-the-Confusion-of-an-Exam
      • 2305-Fair-Distribution-of-Cookies
      • 002616-Minimize-the-Maximum-Difference-of-Pairs
      • 00802-Find-Eventual-Safe-States
    • 中文版解题
      • 1147.longest-chunked-palindrome-decomposition-cn
      • 1168.optimize-water-distribution-in-a-village-cn
      • 1171.remove-zero-sum-consecutive-nodes-from-linked-list-cn
      • 1177.can-make-palindrome-from-substring-cn
      • 215.kth-largest-element-in-an-array-cn
      • 25.reverse-nodes-in-k-groups-cn
      • 30.substring-with-concatenation-of-all-words-cn
      • 4.median-of-two-sorted-array-cn
      • 460.LFU-cache-cn
      • 474.ones-and-zeros-cn
      • 53.maximum-sum-subarray-cn
      • 79.word-search-cn
  • Readings
    • 2020
      • Design-Data-Intensive-Application
      • 亲爱的提奥
      • 理想国
      • 贫穷的本质
Powered by GitBook
On this page
  • Problem
  • Problem Description
  • Solution
  • Complexity Analysis
  • Key Points
  • Code (Java/Python3)

Was this helpful?

  1. Leetcode
  2. English Solution

1171.remove-zero-sum-consecutive-nodes-from-linked-list-en

Previous1168.optimize-water-distribution-in-a-village-enNext1177.can-make-palindrome-from-substring-en

Last updated 5 years ago

Was this helpful?

Problem

Problem Description

Given the head of a linked list, we repeatedly delete consecutive sequences of nodes that sum to 0 until there are no such sequences.

After doing so, return the head of the final linked list.  You may return any such answer.

(Note that in the examples below, all sequences are serializations of ListNode objects.)

Example 1:

Input: head = [1,2,-3,3,1]
Output: [3,1]
Note: The answer [1,2,1] would also be accepted.
Example 2:

Input: head = [1,2,3,-3,4]
Output: [1,2,4]
Example 3:

Input: head = [1,2,3,-3,-2]
Output: [1]

Constraints:

The given linked list will contain between 1 and 1000 nodes.
Each node in the linked list has -1000 <= node.val <= 1000.

Solution

This problem is typical prefixSum problem, if prefixSum seen, then all elements between two same prefixSum sum = 0.

Here, we can use HashMap, key as prefixSum, value as current Node, if prefixSum already seen in HashMap, then delete all nodes between two same prefixSum. and also delete relative prefixSum in HashMap. (We already removed nodes from head, so that remove cumulative sum with those nodes. otherwise, it is incorrect.)

Here, we use dummy node, ListNode(0), (as shown in below pic), as initiating HashMap, put(0, ListNode(0)) into HashMap, iterate ListNode,

  • if prefixSum in HashMap, delete all nodes

  • if prefixSum not in HashMap, put prefixSum and currNode into HashMap. map.put(prefixSum, currNode).

举例:head = [1, 2, 3, -2, -3, 9]

Complexity Analysis

  • Time Complexity: O(n) - n is number of ListNode

  • Space Complexity: O(n) - HashMap

Key Points

  • calculate prefixSum

  • use HashMap keep track all prefixSum and currNode.

  • Initialize HashMap,put0 (here usedummy node)

  • Delete all Nodes between two same prefixSum, delete related prefixSum from HashMap.

  • return dummy.next;.

Code (Java/Python3)

Java Code

class RemoveZeroInLinkedList {
  public static ListNode removeZeroSumSublists(ListNode head) {
      ListNode dummy = new ListNode(0);
      ListNode currNode = dummy;
      dummy.next = head;
      int prefixSum = 0;
      Map<Integer, ListNode> prefixSumMap = new HashMap<>();
      while (currNode != null) {
        prefixSum += currNode.val;
        // seen prefixSum in HashMap, remove all nodes and relative prefixSum from HashMap
        if (prefixSumMap.containsKey(prefixSum)) {
          currNode = prefixSumMap.get(prefixSum).next;
          int sum = prefixSum + currNode.val;
          while (sum != prefixSum) {
            prefixSumMap.remove(sum);
            currNode = currNode.next;
            sum += currNode.val;
          }
          prefixSumMap.get(prefixSum).next = currNode.next;
        } else {
          // prefixSum not seen in HashMap, add into HashMap
          prefixSumMap.put(prefixSum, currNode);
        }
        currNode = currNode.next;
      }
      return dummy.next;
  }
}

Python3 code

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def removeZeroSumSublists(self, head: ListNode) -> ListNode:
        dummy = curr_node = ListNode(0)
        dummy.next = head
        prefix_sum = 0
        prefix_sum_map = {}
        while curr_node:
            prefix_sum += curr_node.val
            if prefix_sum in prefix_sum_map:
                curr_node = prefix_sum_map.get(prefix_sum).next
                sum = prefix_sum + curr_node.val
                while sum != prefix_sum and sum in prefix_sum_map:
                    del prefix_sum_map[sum]
                    curr_node = curr_node.next
                    sum += curr_node.val
                prefix_sum_map[prefix_sum].next = curr_node.next
            else:
                prefix_sum_map[prefix_sum] = curr_node
            curr_node = curr_node.next
        return dummy.next
https://leetcode.com/problems/remove-zero-sum-consecutive-nodes-from-linked-list/
1171 example remove zero