Update Remove Zero Sum Consecutive Nodes from Linked List.java

varunu28 · web-flow · commit c0bd347683b4 · 2024-03-11T17:46:11.000-07:00
diff --git a/Medium/Remove Zero Sum Consecutive Nodes from Linked List.java b/Medium/Remove Zero Sum Consecutive Nodes from Linked List.java
@@ -9,25 +9,24 @@
  * }
  */
 class Solution {
-  public ListNode removeZeroSumSublists(ListNode head) {
-    ListNode dummy = new ListNode(0);
-    ListNode curr = head;
-    dummy.next = head;
-    Map<Integer, ListNode> map = new HashMap<>();
-    int prefixSum = 0;
-    map.put(0, dummy);
-    while (curr != null) {
-      prefixSum += curr.val;
-      map.put(prefixSum, curr);
-      curr = curr.next;
+    public ListNode removeZeroSumSublists(ListNode head) {
+        ListNode dummy = new ListNode(0, head);
+        ListNode curr = dummy;
+        int prefixSum = 0;
+        Map<Integer, ListNode> map = new HashMap<>();
+        map.put(0, curr);
+        while (curr != null) {
+            prefixSum += curr.val;
+            map.put(prefixSum, curr);
+            curr = curr.next;
+        }
+        prefixSum = 0;
+        curr = dummy;
+        while (curr != null) {
+            prefixSum += curr.val;
+            curr.next = map.get(prefixSum).next;
+            curr = curr.next;
+        }
+        return dummy.next;
     }
-    prefixSum = 0;
-    curr = dummy;
-    while (curr != null) {
-      prefixSum += curr.val;
-      curr.next = map.get(prefixSum).next;
-      curr = curr.next;
-    }
-    return dummy.next;
-  }
 }
