Code Ganker: Sort List -- LeetCode

2014年3月12日星期三

Sort List -- LeetCode

原题链接: http://oj.leetcode.com/problems/sort-list/
这道题跟Insertion Sort List类似,要求我们用O(nlogn)算法对链表进行排序,但是并没有要求用哪一种排序算法,我们可以使用归并排序,快速排序,堆排序等满足要求的方法来实现。对于这道题比较容易想到的是归并排序,因为我们已经做过Merge Two Sorted Lists,这是归并排序的一个subroutine。剩下我们需要做的就是每次找到中点,然后对于左右进行递归,最后用Merge Two Sorted Lists把他们合并起来。代码如下:
public ListNode sortList(ListNode head) {
    return mergeSort(head);
}
private ListNode mergeSort(ListNode head)
{
    if(head == null || head.next == null)
        return head;
    ListNode walker = head;
    ListNode runner = head;
    while(runner.next!=null && runner.next.next!=null)
    {
        walker = walker.next;
        runner = runner.next.next;
    }
    ListNode head2 = walker.next;
    walker.next = null;
    ListNode head1 = head;
    head1 = mergeSort(head1);
    head2 = mergeSort(head2);
    return merge(head1, head2);
}
private ListNode merge(ListNode head1, ListNode head2)
{
    ListNode helper = new ListNode(0);
    helper.next = head1;
    ListNode pre = helper;
    while(head1!=null && head2!=null)
    {
        if(head1.val<head2.val)
        {
            head1 = head1.next;
        }
        else
        {
            ListNode next = head2.next;
            head2.next = pre.next;
            pre.next = head2;
            head2 = next;
        }
        pre = pre.next;
    }
    if(head2!=null)
    {
        pre.next = head2;
    }
    return helper.next;
}
不过用归并排序有个问题就是这里如果把栈空间算上的话还是需要O(logn)的空间的。对于其他排序算法,用兴趣的同学可以实现一下哈。
排序是面试中比较基础的一个主题,所以对于各种常见的排序算法大家还是要熟悉,不了解的朋友可以参见排序算法 - Wiki。特别是算法的原理,很多题目虽然没有直接考察排序的实现,但是用到了其中的思想,比如非常经典的topK问题,就用到了快速排序的原理,关于这个问题在Median of Two Sorted Arrays中有提到,有兴趣的朋友可以看看。

2 条评论:

  1. hi, 大牛,为什么line 16 (according to http://blog.csdn.net/linhuanmars/article/details/21133949) 要有walker.next = null;? 如果没有这个会有stackOverflow,但是为什么要有呢?多谢!

    回复删除
    回复
    1. 如果没有这个会造成链表循环,搞个例子试试就知道了哈~

      删除