题目描述(中等难度)
要求时间复杂度为 O(nlogn)
,最常用的就是归并排序了。
解法一
归并排序需要一个辅助方法,也就是对两个有序链表进行合并,在 21 题 已经讨论过。
至于归并排序的思想,这里就不多讲了,本科的时候用 Scratch
做过一个演示视频,感兴趣的可以参考 这里,哈哈。
那就直接放代码了。因为归并排序是一半一半的进行,所以需要找到中点。最常用的方法就是快慢指针去找中点了。
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode fast = dummy;
ListNode slow = dummy;
while (fast != null && fast.next != null) {
slow = slow.next;
fast = fast.next.next;
}
上边的代码我加了一个 dummy
指针,就是想当节点个数是偶数的时候,让 slow
刚好指向前边一半节点的最后一个节点,也就是下边的状态。
1 2 3 4
^ ^
slow fast
如果 slow
和 fast
都从 head
开始走,那么当 fast
结束的时候,slow
就会走到后边一半节点的开头了。
当然除了上边的方法,在 这里 看到,还可以加一个 pre
指针,让它一直指向 slow
的前一个即可。
// step 1. cut the list to two halves
ListNode prev = null, slow = head, fast = head;
while (fast != null && fast.next != null) {
prev = slow;
slow = slow.next;
fast = fast.next.next;
}
他们的目的都是一样的,就是为了方便的把两个链表平均分开。
public ListNode sortList(ListNode head) {
return mergeSort(head);
}
private ListNode mergeSort(ListNode head) {
if (head == null || head.next == null) {
return head;
}
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode fast = dummy;
ListNode slow = dummy;
//快慢指针找中点
while (fast != null && fast.next != null) {
slow = slow.next;
fast = fast.next.next;
}
ListNode head2 = slow.next;
slow.next = null;
head = mergeSort(head);
head2 = mergeSort(head2);
return merge(head, head2);
}
private ListNode merge(ListNode head1, ListNode head2) {
ListNode dummy = new ListNode(0);
ListNode tail = dummy;
while (head1 != null && head2 != null) {
if (head1.val < head2.val) {
tail.next = head1;
tail = tail.next;
head1 = head1.next;
} else {
tail.next = head2;
tail = tail.next;
head2 = head2.next;
}
}
if (head1 != null) {
tail.next = head1;
}
if (head2 != null) {
tail.next = head2;
}
return dummy.next;
}
当然严格的说,上边的解法空间复杂度并不是 O(1)
,因为递归过程中压栈是需要消耗空间的,每次取一半,所以空间复杂度是 O(log(n))
。
递归可以去改写成迭代的形式,也就是自底向上的走,就可以省去压栈的空间,空间复杂度从而达到 O(1)
,详细的可以参考 这里-with-o(1)-space-complextity-and-o(nlgn)-time-complextity) 。
总
和 147 题 一样,主要还是考察对链表的理解和排序算法的实现。