您的位置:首页 > 其它

链表的二路归并排序 Sort List

2014-03-12 13:43 465 查看
对于链表的排序问题,想找一种时间复杂度为O(nlogn),空间复杂度O(1)的算法。该问题源自于Leetcode,Sort a linked list in O(n log n)
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:链表的排序用快速排序应该也可以做到。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: