Leetcode 面试题 02.04. 分割链表

编写程序以 x 为基准分割链表,使得所有小于 x 的节点排在大于或等于 x 的节点之前。如果链表中包含 x,x 只需出现在小于 x 的元素之后(如下所示)。分割元素 x 只需处于“右半部分”即可,其不需要被置于左右两部分之间。

示例:

输入: head = 3->5->8->5->10->2->1, x = 5

输出: 3->1->2->10->5->5->8

**难度**: Medium

**标签**: 链表、 双指针、


# -*- coding: utf-8 -*-
# @Author  : LG

"""
执行用时:44 ms, 在所有 Python3 提交中击败了69.92% 的用户
内存消耗:13.7 MB, 在所有 Python3 提交中击败了100.00% 的用户

解题思路:
    使用双指针,一个指针遍历,一个指针指向链头。
    若当前节点值小于阈值,则将当前节点挪到链头。
"""

# 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:
        p, q = head, head
        if not head:
            return head
        while q.next:
            if q.next.val < x:
                r = q.next
                q.next = q.next.next
                r.next = p
                p = r
            else:
                q = q.next
        return p