leetcode刷題（77）—— 148. 排序連結清單

在 O(n log n) 時間複雜度和常數級空間複雜度下，對連結清單進行排序。

示例 1:

輸入: 4->2->1->3

輸出: 1->2->3->4

示例 2:

輸入: -1->5->3->4->0

輸出: -1->0->3->4->5

最開始笨拙的想法

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode sortList(ListNode head) {
           PriorityQueue<Integer> heap =
                new PriorityQueue<Integer>(new Comparator<Integer>() {
                    @Override
                    public int compare(Integer n1, Integer n2) {
                        return n1 - n2;
                    }
                });

        while (head != null) {
            heap.add(head.val);
            head = head.next;
        }
        ListNode pre = new ListNode(0);
        ListNode begin = new ListNode(0);
        pre = begin;
        while (!heap.isEmpty()) {
            ListNode node = new ListNode(heap.poll());
            begin.next = node;
            begin = node;
        }
        return pre.next;
    }
}           

複制

public static ListNode sortList(ListNode head) {
        // 1、遞歸結束條件
        if (head == null || head.next == null) {
            return head;
        }

        // 2、找到連結清單中間節點并斷開連結清單 & 遞歸下探
        ListNode midNode = middleNode(head);
        ListNode rightHead = midNode.next;
        midNode.next = null;

        ListNode left = sortList(head);
        ListNode right = sortList(rightHead);

        // 3、目前層業務操作（合并有序連結清單）
        return mergeTwoLists(left, right);
    }

    //  找到連結清單中間節點（876. 連結清單的中間結點）
    private static ListNode middleNode(ListNode head) {
        if (head == null || head.next == null) {
            return head;
        }
        ListNode slow = head;
        ListNode fast = head.next.next;

        while (fast != null && fast.next != null) {
            slow = slow.next;
            fast = fast.next.next;
        }

        return slow;
    }

    // 合并兩個有序連結清單（21. 合并兩個有序連結清單）
    private static ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode sentry = new ListNode(-1);
        ListNode curr = sentry;

        while(l1 != null && l2 != null) {
            if(l1.val < l2.val) {
                curr.next = l1;
                l1 = l1.next;
            } else {
                curr.next = l2;
                l2 = l2.next;
            }

            curr = curr.next;
        }

        curr.next = l1 != null ? l1 : l2;
        return sentry.next;
    }           

複制