Topic:
给你一个链表和一个特定值 x ,请你对链表进行分隔,使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
Example:
输入:head = 1->4->3->2->5->2, x = 3
输出:1->2->2->4->3->5
思路:
整体的思路就是将链表按原序拆分为小的和大的两个链表,最后合并
首先先设置两个哑节点(便于边界的判断)
之后遍历整个链表
若head的val小于x,则small的下一位变为head
small指针转换为下一位(此时head的val)
同理
若head的val大于或等于于x,则big的下一位变为head
big指针指向下一位(此时head的val)
这样两个链表已经按照原来的顺序
分别将小的保存为了small
大的保存为了big
遍历结束后
将 big 的 next 指针清空(防止指向原来初始链表的节点)
同时将 small 的 next 指向L2的next 的节点(big链表真正的头节点)
最后返回L1的next指针
Code:
class Solution:
def partition(self, head: ListNode, x: int) -> ListNode:
L1 = small = ListNode(0)
L2 = big = ListNode(0)
while head:
if head.val < x:
small.next = head
small = small.next
else:
big.next = head
big = big.next
head = head.next
big.next = None
small.next = L2.next
return L1.next
Result: