re-solve "86. Partition List"

lancelote · Nov 2, 2024 · d03b226 · d03b226
1 parent b6628cc
commit d03b226
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/src/partition_list.py b/src/partition_list.py
@@ -3,19 +3,18 @@
 
 class Solution:
     def partition(self, head: ListNode | None, x: int) -> ListNode | None:
-        less_head = less = ListNode()
-        more_head = more = ListNode()
+        dummy_ls = ls = ListNode()
+        dummy_gt = gt = ListNode()
 
         while head:
             if head.val < x:
-                less.next = head
-                less = less.next
+                ls.next = head
+                ls = ls.next
             else:
-                more.next = head
-                more = more.next
+                gt.next = head
+                gt = gt.next
             head = head.next
 
-        less.next = more_head.next
-        more.next = None
-
-        return less_head.next
+        gt.next = None
+        ls.next = dummy_gt.next
+        return dummy_ls.next