Merge pull request #2972 from coopers/0023

Update 0023-merge-k-sorted-lists.py

Author: felivalencia3

python/0023-merge-k-sorted-lists.py

@@ -18,19 +18,16 @@ def mergeKLists(self, lists: List[ListNode]) -> ListNode:
         return lists[0]
 
     def mergeList(self, l1, l2):
-        dummy = ListNode()
-        tail = dummy
+        dummy = node = ListNode()
 
         while l1 and l2:
             if l1.val < l2.val:
-                tail.next = l1
+                node.next = l1
                 l1 = l1.next
             else:
-                tail.next = l2
+                node.next = l2
                 l2 = l2.next
-            tail = tail.next
-        if l1:
-            tail.next = l1
-        if l2:
-            tail.next = l2
+            node = node.next
+
+        node.next = l1 or l2
         return dummy.next