【问题标题】:Python implementation of Mergesort for Linked list doesn't work用于链接列表的合并排序的 Python 实现不起作用
【发布时间】:2017-01-13 05:26:59
【问题描述】:

我无法在 Python 中为 Linked Lists 在任何地方找到简单的 Merge Sort 实现。这是我尝试过的:

单链表定义:

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None 

合并排序实现:

def mergeSortLinkedList(A):
    # Base case length of 0 or 1:
    if A == None or A.next == None:
        return A

    leftHalf, rightHalf = splitTheList(A)

    mergeSortLinkedList(leftHalf)
    mergeSortLinkedList(rightHalf)

    # The above two lines should be modified to the following. Thanks to the answers.

    # leftHalf  = mergeSortLinkedList(leftHalf)
    # rightHalf = mergeSortLinkedList(rightHalf)

    return mergeTheLists(leftHalf, rightHalf)

拆分:

def splitTheList(sourceList):
    if sourceList == None or sourceList.next == None:
        leftHalf = sourceList
        rightHalf = None

        return leftHalf, rightHalf

    else:
        midPointer = sourceList
        frontRunner = sourceList.next
        # totalLength += 1        - This is unnecessary

        while frontRunner != None:
            frontRunner = frontRunner.next

            if frontRunner != None:
                frontRunner = frontRunner.next
                midPointer = midPointer.next

    leftHalf = sourceList
    rightHalf = midPointer.next
    midPointer.next = None

    return leftHalf, rightHalf

合并:

def mergeTheLists(leftHalf, rightHalf):
    fake_head = ListNode(None)
    curr = fake_head

    while leftHalf and rightHalf:
        if leftHalf.val < rightHalf.val:
            curr.next = leftHalf
            leftHalf = leftHalf.next

        else:
            curr.next = rightHalf
            rightHalf = rightHalf.next

        curr = curr.next

    if leftHalf == None:
        curr.next = rightHalf

    elif rightHalf == None:
        curr.next = leftHalf

    return fake_head.next

数据:

# Node A:
nodeA1 = ListNode(2)

nodeA2 = ListNode(1)
nodeA1.next = nodeA2

nodeA3 = ListNode(9)
nodeA2.next = nodeA3

nodeA4 = ListNode(3)
nodeA3.next = nodeA4

# Node C:
nodeC1 = ListNode(5)
nodeA4.next = nodeC1

nodeC2 = ListNode(6)
nodeC1.next = nodeC2

nodeC3 = ListNode(4)
nodeC2.next = nodeC3

nodeC4 = ListNode(5)
nodeC3.next = nodeC4

调用mergeSortLinkedList(nodeA1) 时的预期输出:

1 2 3 4 5 5 6 9

我得到以下信息:

2 5 6 9

我无法弄清楚小姐在哪里。请帮忙。

【问题讨论】:

  • 如果您有兴趣,还有一种更简单的替代自下而上方法,它使用对节点的引用数组,并且不必扫描列表将它们分成两半。 wiki article

标签: python algorithm sorting linked-list mergesort


【解决方案1】:

您不使用递归调用的返回值。代码应该是:

def mergeSortLinkedList(A):
    if A is None or A.next is None:
        return A

    leftHalf, rightHalf = splitTheList(A)

    left = mergeSortLinkedList(leftHalf)
    right = mergeSortLinkedList(rightHalf)

    return mergeTheLists(left, right)

函数调用后,某些情况下参数不指向排序列表的头部。

代码中的下一个错误是使用了未定义的变量 totalLength。

【讨论】:

    猜你喜欢
    • 2021-06-04
    • 1970-01-01
    • 2019-11-24
    • 1970-01-01
    • 1970-01-01
    • 2022-01-03
    • 2010-09-05
    • 2022-01-10
    相关资源
    最近更新 更多