前置知识:理解链表及其基本调整

建议:做过这个题的同学跳过

86.分隔链表

给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。

你应当 保留 两个分区中每个节点的初始相对位置(稳定性)

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
 
func partition(head *ListNode, x int) *ListNode {
	var (
		leftHead  *ListNode
		leftTail  *ListNode
		rightHead *ListNode
		rightTail *ListNode
	)
 
	for head != nil {
		if head.Val < x {
			if leftTail == nil {
				leftHead = head
			} else {
				leftTail.Next = head
			}
			leftTail = head
		} else {
			if rightTail == nil {
				rightHead = head
			} else {
				rightTail.Next = head
			}
			rightTail = head
		}
		head = head.Next
	}
 
	if rightTail != nil { // 清除原指向
		rightTail.Next = nil
	}
	if leftHead == nil { // 没有 <x 的元素
		return rightHead
	}
	leftTail.Next = rightHead // 指向正确值
	return leftHead
}

coding 能力