-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMerge_k_Sorted_Lists.py
34 lines (33 loc) · 1.02 KB
/
Merge_k_Sorted_Lists.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def mergeKLists(self, lists: List[Optional[ListNode]]) -> Optional[ListNode]:
if not lists or len(lists)==0:
return None
while len(lists)>1:
mergedList = []
for i in range(0,len(lists),2):
l1 = lists[i]
l2 = lists[i+1] if (i+1)<len(lists) else None
mergedList.append(self.mergeList(l1,l2))
lists = mergedList
return lists[0]
def mergeList(self,l1,l2):
dummy = ListNode()
tail = dummy
while l1 and l2:
if l1.val<l2.val:
tail.next = l1
l1 = l1.next
else:
tail.next = l2
l2 = l2.next
tail = tail.next
if l1:
tail.next = l1
if l2:
tail.next = l2
return dummy.next