用 Python 合并两个已排序列表
假设我们有两个已排序列表 A 和 B。我们必须合并它们并只生成一个已排序列表 C。列表的大小可能不同。
例如,假设 A = [1,2,4,7] 且 B = [1,3,4,5,6,8],则合并后的列表 C 为 [1,1,2,3,4,4,5,6,7,8]
我们使用递归解决这个问题。所以函数将如下工作 -
- 假设 merge() 函数的列表 A 和 B
- 如果 A 为空,则返回 B,如果 B 为空,则返回 A
- 如果 A 中的值 <= B 中的值,则 A.next = merge(A.next, B) 并返回 A
- 否则,则 B.next = merge(A, B.next) 并返回 B
让我们看看实现以更好地理解
示例(Python)
class ListNode: def __init__(self, data, next = None): self.val = data self.next = next def make_list(elements): head = ListNode(elements[0]) for element in elements[1:]: ptr = head while ptr.next: ptr = ptr.next ptr.next = ListNode(element) return head def print_list(head): ptr = head print('[', end = "") while ptr: print(ptr.val, end = ", ") ptr = ptr.next print(']') class Solution: def mergeTwoLists(self, l1, l2): """ :type l1: ListNode :type l2: ListNode :rtype: ListNode """ if not l1: return l2 if not l2: return l1 if(l1.val<=l2.val): l1.next = self.mergeTwoLists(l1.next,l2) return l1 else: l2.next = self.mergeTwoLists(l1,l2.next) return l2 head1 = make_list([1,2,4,7]) head2 = make_list([1,3,4,5,6,8]) ob1 = Solution() head3 = ob1.mergeTwoLists(head1,head2) print_list(head3)
输入
head1 = make_list([1,2,4,7]) head2 = make_list([1,3,4,5,6,8])
输出
[1, 1, 2, 3, 4, 4, 5, 6, 7, 8, ]
广告