前置知识:理解链表及其基本调整
建议:做过这个题的同学跳过
给你一个链表的头节点 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
}