LeetCode 148 Sort List

2016-08-28  本文已影响209人  ShuiLocked

LeetCode 148 Sort List

Sort a linked list in O(n log n) time using constant space complexity.

对于linked list而言,除了insertion sort比较容易实现以外,剩下o(nlogn)的算法中,merge sort比较容易实现。

大致思路:用快慢指针,每次都找到list的中点,然后递归对前后两段分别sort,再merge排序后的前后两段list。

也即divide and conquer的方法:Each time divided the given list into two sub list. Merge sub list after bottom case return.

我一开始纠结在,通过什么方式divide???
其实只要分别对前半段和后半段list分别调用sort函数,传入head与slow.next,其实等价于实现了divide!!!对sort好的前半段后半段再merge即可。

而sort每一段list可以分为两种情况:
list为null或者单节点时,直接返回。
list>=2个node时,将其分成前后半段,各自sort后,合并。
所以最基础的情况,是从2个node开始合并,然后再越并越长。

这里要注意一个trick

操作slow与fast时,到底用slow还是slow.next代表后半段的开始?!

如果用slow代表后半段的开始,那还需要一个prev node,每次记录slow的前一个节点,这样才能将前半段的next最后赋值为null,从而cut前半段与后半段。

而用slow.next代表后半段第一个node,则直接slow.next=null,就可以完成cut!!!

代码:

public class Solution {
    public ListNode sortList(ListNode head) {
        // If list is null or only has 1 node, then return
        if (head == null || head.next == null) return head;
        
        // Use slow & fast pointer to find the middle of list
        ListNode slow = head, fast = head.next.next;
        while (fast != null && fast.next != null) {
            slow = slow.next;
            fast = fast.next.next;
        }
        // slow.next points to the middle!!! 
        // Break up the list at slow.next by assigning slow.next = null
        ListNode h2 = sortList(slow.next);
        slow.next = null;
        return merge(sortList(head), h2);
    }
    
    public ListNode merge(ListNode p1, ListNode p2) {
        // How to choose which node as the first node?!
        // To deal with the corner case of the first node, we use a dummy node
        // So the assign of all nodes can be implemented by using next pointer!!!
        ListNode dummy = new ListNode(Integer.MIN_VALUE);
        ListNode c = dummy;
        while (p1 != null && p2 != null) {
            if (p1.val < p2.val) {
                c.next = p1;
                p1 = p1.next;    
            } else { 
                c.next = p2;
                p2 = p2.next;
            }
            c = c.next;
        }
        
        if (p1 != null) c.next = p1;
        if (p2 != null) c.next = p2;
        
        return dummy.next;
    }
    
    
}
上一篇下一篇

猜你喜欢

热点阅读