86. 分隔链表
原始题目链接:https://leetcode-cn.com/problems/partition-list/
给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。你应当 保留 两个分区中每个节点的初始相对位置。
解题思路:
以x大小为界,将原始链表分为大小两个链表,然后再拼接一起即可,具体实现看代码及注释。
代码实现:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def partition(self, head: ListNode, x: int) -> ListNode:
# 判空
if head is None:
return head
# 申请两个节点,即两个dummy
dummy1 = ListNode(-1)
dummy2 = ListNode(-1)
# 分别遍历指向head较小的节点和较大的节点
# 形成以x为界限的大小两个链表
small = dummy1
large = dummy2
while head:
if head.val < x:
small.next = head
small = small.next
else:
large.next = head
large = large.next
head = head.next
# 将大、小链表进行拼接
# 小的链表的尾节点的next指针指向大链表的头结点
small.next = dummy2.next
# 大链表的尾节点的next指针指向空,表示整个拼接后链表的尾节点
large.next = None
# 返回小链表的头结点就是最后拼接后链表的头结点
return dummy1.next
参考文献:
https://leetcode-cn.com/problems/partition-list/solution/fen-ge-lian-biao-by-leetcode-solution-7ade/