上一篇文章用数组实现了二路归并排序,核心在于要对每次减半的元素进行实质性的改变,在数组中我们用临时数组记录每次merge的结果,在链表中用归并排序更加直接,因为我们操作的是结点的指针。
题目描述
难度:中等
Sort a linked list in O(n log n) time using constant space complexity.
Example 1:
Input: 4->2->1->3
Output: 1->2->3->4
Example 2:
Input: -1->5->3->4->0
Output: -1->0->3->4->5
题解
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* dummy = new ListNode(-1);
ListNode* cur = dummy;
while (l1 && l2) {
if (l1->val < l2->val) {
cur->next = l1;
cur = l1;
l1 = l1->next;
} else {
cur->next = l2;
cur = l2;
l2 = l2->next;
}
}
if (l1) {
cur->next = l1;
}
if (l2) {
cur->next = l2;
}
return dummy->next;
}
ListNode* sortList(ListNode* head) {
// 归并排序,找到中间结点的方法是:快慢指针
if (!head || !head->next) {
return head;
}
ListNode* fast = head, *slow = head;
// 奇数个元素时,slow结束时指向中间元素
// 偶数个元素时,slow结束时指向前半段的最后一个元素,所以需要slow = slow->next
while (fast->next && fast->next->next) {
fast = fast->next->next;
slow = slow->next;
}
// 断开成两部分
fast = slow;
slow = slow->next;
fast->next = nullptr;
ListNode* l1 = sortList(head);
ListNode* l2 = sortList(slow);
return mergeTwoLists(l1, l2);
}
};
在数组中计算中间点的方法是直接用下标计算,如果能够解决链表中的这个问题,则归并排序就很自然了。
回想一下快慢指针的功效,让快指针比慢指针速度快一倍,那么等快指针都跑完了,则慢指针才跑一半。
链表本身的长度是有限的,所以这个方法很可行。
找到中间点以后,需要把链表打断,调用merge即可,关于递归,在前文讲了一些思考(https://blog.csdn.net/u011240016/article/details/87903240),这里不再赘述。
PS. 想起实现这段代码时更新cur = l1
时位置放在了l1 = l1->next
后面导致的空指针问题。对于指针问题,需要特别注意操作的顺序,一点点的顺序错位,都会导致完全不同的结果。
END.