对于链表的排序问题,想找一种时间复杂度为O(nlogn),空间复杂度O(1)的算法。该问题源自于Leetcode,Sort a linked list inO(nlogn) time using constant space complexity.
首先想到了归并排序,因为链表的归并不需要辅助空间。
思路:和数组的二路归并基本一致。
所不同的是:
1、要遍历一遍才能找到中间位置。
2、首尾位置都要记下来,因为链表不是自然相邻的。所以排序后返回时要返回首尾结点的新指针。
3、遍历查找时也要注意链表结尾判定可能不再是NULL。
/**
* 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 == NULL)
return NULL;
ListNode *end = head;
while(end->next != NULL)
end = end->next;
sort(head, end);
return head;
}
void sort(ListNode* &left, ListNode* &right)
{
if(left == right)
return;
//寻找中间位置
ListNode *p1, *p2;
p1 = p2 = left;
while(p2 != right && p2->next != right)
{
p1 = p1->next;
p2 = p2->next->next;
}
p2 = p1->next;
//递归分治
sort(left, p1);
sort(p2, right);
p1->next = NULL;
right->next = NULL;
//归并
ListNode *new_start,*new_end, *tmp;
if(left->val < p2->val)
{
new_start = left;
left = left->next;
}
else
{
new_start = p2;
p2 = p2->next;
}
tmp = new_start;
while(left != NULL && p2 != NULL)
{
if(left->val < p2->val)
{
tmp->next = left;
tmp = left;
left = left->next;
}
else
{
tmp->next = p2;
tmp = p2;
p2 = p2->next;
}
}
if(left != NULL)
{
tmp->next = left;
new_end = p1;
}
else
{
tmp->next = p2;
new_end = right;
}
//返回新链表头和链表尾
left = new_start;
right = new_end;
right->next = NULL;
}
};
PS:链表的排序用快速排序应该也可以做到。