leetcode刷题python之分割链表

题目：给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

思路：将小于x的值放到左侧，大于等于x的值放到右侧链表，再把两个链表拼接到一起。right.next = None # 必须有这句话，如果没有这句话切断末尾的话，right很有可能会指向head中的某个值，这个值又会以原来head中的顺序指向right，形成死循环

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def partition(self, head: ListNode, x: int) -> ListNode:
        left = ListNode(0)
        right = ListNode(0)
        res = left
        mid = right
        while head:
            if head.val < x:
                left.next = head
                left = left.next
            else:
                right.next = head
                right = right.next
            head = head.next
        left.next = mid.next
        right.next = None  # 必须有这句话，如果没有这句话切断末尾的话，right很有可能会指向head中的某个值，这个值又会以原来head中的顺序指向right，形成死循环
        return res.next

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。