题目来自leetcode:Sort a linked list in O(n log n) time using constant space complexity.
快速排序平均复杂度是O(nlogn),但最坏的时候是O(n^2),不符合要求。考虑采用归并排序,注意空间复杂度为O(1)。
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
ListNode * mergeList(ListNode *head1,ListNode *head2)
{
if(NULL == head1)
return head2;
if(NULL == head2)
return head1;
ListNode * p_head = NULL;
if(head1->val <= head2->val){
p_head = head1;
head1 = head1->next;
}
else{
p_head = head2;
head2 = head2->next;
}
ListNode * p_tmp = p_head;
while(head1 && head2)
{
if(head1->val <= head2->val){
p_tmp->next = head1;
head1 = head1->next;
}
else{
p_tmp ->next = head2;
head2 = head2->next;
}
p_tmp = p_tmp->next;
}
if(head1)
p_tmp->next = head1;
else
p_tmp->next = head2;
return p_head;
}
ListNode * get_mid(ListNode *head){
ListNode *fast = head->next;
while(fast && fast->next){
head = head->next;
fast = fast->next->next;
}
return head;
}
ListNode *sortList(ListNode *head) {
if( NULL==head || NULL==head->next)
return head;
ListNode *p_mid = get_mid(head);
ListNode * mid_next = p_mid->next;
p_mid->next = NULL;
return mergeList(sortList(head),sortList(mid_next));
}