[leetcode]Python实现-147.对链表进行插入排序

147.对链表进行插入排序

描述

对链表进行插入排序。
插入排序的动画演示如上。从第一个元素开始,该链表可以被认为已经部分排序(用黑色表示)。
每次迭代时,从输入数据中移除一个元素(用红色表示),并原地将其插入到已排好序的链表中。

插入排序算法
插入排序是迭代的,每次只移动一个元素,直到所有元素可以形成一个有序的输出列表。
每次迭代中,插入排序只从输入数据中移除一个待排序的元素,找到它在序列中适当的位置,并将其插入。
重复直到所有输入数据插入完为止。

示例

输入: 4->2->1->3
输出: 1->2->3->4

输入: -1->5->3->4->0
输出: -1->0->3->4->5

思路:设置一个新的链表。遍历旧链表,每一个节点都作为一个新节点,按序插入新链表中。

        new_s = ListNode(None)
        new_s.next = ListNode(head.val)
        cur = head.next
        while cur:
            insert_node = ListNode(cur.val)
            new_cur = new_s.next
            pre = new_s
            while new_cur:
                if new_cur.val >= insert_node.val:
                    insert_node.next = new_cur
                    pre.next = insert_node
                    break
                elif new_cur.next is None:
                    new_cur.next = insert_node
                    break
                pre = new_cur
                new_cur = new_cur.next
            cur = cur.next
        return new_s.next

超时。
修改

        p = new_s = ListNode(None)
        new_s.next = head
        cur = head
        while cur and cur.next:
            val = cur.next.val
            if val > cur.val:
                cur = cur.next
                continue
            if p.next.val > val:
                p = new_s
            while p.next.val < val:
                p = p.next
            new = cur.next
            cur.next = new.next
            new.next = p.next
            p.next = new
        return new_s.next

还有一种走捷径的方法,那就是遍历一次链表,将所有节点的值拿出来进行排序,然后再根据排序后的结果生成一个新链表。

        h = head
        res = []
        while h:
            res.append(h.val)
            h = h.next
        res.sort()
        h = head
        for i in res:
            h.val = i
            h = h.next
        return head

你可能感兴趣的:(leetcode,Python)