码迷,mamicode.com
首页 > 其他好文 > 详细

【Sort List】cpp

时间:2015-05-18 22:34:18      阅读:119      评论:0      收藏:0      [点我收藏+]

标签:

题目:

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

代码:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* sortList(ListNode* head) {
        if ( !head || !head->next ) return head;
        ListNode dummy(-1);
        dummy.next = head;
        ListNode *p1=&dummy, *p2=&dummy;
        while ( p2 && p2->next && p2->next->next )
        {
            p1 = p1->next;
            p2 = p2->next->next;
        }
        ListNode *h1 =  Solution::sortList(p1->next);
        p1->next = NULL;
        ListNode *h2 =  Solution::sortList(dummy.next);
        return Solution::mergeTwo(h1, h2);
    }
    static ListNode* mergeTwo(ListNode *h1, ListNode *h2)
    {
        ListNode dummy(-1);
        ListNode *p = &dummy;
        while ( h1 && h2 )
        {
            if ( h1->val<h2->val )
            {
                p->next = h1;
                h1 = h1->next;
            }
            else
            {
                p->next = h2;
                h2 = h2->next;
            }
            p = p->next;
        }
        p->next = h1 ? h1 : h2;
        return dummy.next;
    }
};

tips:

单链表时间要求O(nlongn) 且const extra space,可以选择归并排序(另,双向链表适合用快速排序)

第一次没有AC,原因是少考虑一种返回条件,即“head只有一个元素的时候需要直接返回”,修改之后第二次AC了。

【Sort List】cpp

标签:

原文地址:http://www.cnblogs.com/xbf9xbf/p/4513027.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!