leetcode-148:排序链表

题目描述

给你链表的头结点 head ，请将其按 升序 排列并返回 排序后的链表 。

进阶：

你可以在 O(n log n) 时间复杂度和常数级空间复杂度下，对链表进行排序吗？

示例 1：

输入：head = [4,2,1,3]
输出：[1,2,3,4]
示例 2：

输入：head = [-1,5,3,4,0]
输出：[-1,0,3,4,5]
示例 3：

输入：head = []
输出：[]

解题思路

• 归并排序
• 快慢指针找链表中点进行拆分
• 将拆分的链表继续进行拆分（递归），直到拆分为单个节点
• 排序合并两个单节点的链表

代码实现

func sortList(head *ListNode) *ListNode {
	return mergeSort(head)
}

func findMiddle(head *ListNode) *ListNode {
	fast := head.Next
	slow := head
	for fast != nil && fast.Next != nil {
		slow = slow.Next
		fast = fast.Next.Next
	}
	return slow
}

func mergeList(l1 *ListNode, l2 *ListNode) *ListNode {
	res := &ListNode{Val: 0}
	ptr := res
	for l1 != nil && l2 != nil {
		if l1.Val > l2.Val {
			ptr.Next = l2
			l2 = l2.Next
		} else {
			ptr.Next = l1
			l1 = l1.Next
		}
		ptr = ptr.Next
	}

	if l1 != nil {
		ptr.Next = l1
	}
	if l2 != nil {
		ptr.Next = l2
	}

	return res.Next
}

func mergeSort(head *ListNode) *ListNode {
	if head == nil || head.Next == nil {
		return head
	}
	mid := findMiddle(head)
	tail := mid.Next
	mid.Next = nil

	l1 := mergeSort(head)
	l2 := mergeSort(tail)
	return mergeList(l1, l2)
}