leetcode专题训练 86. Partition List

这道题是要将小于x的节点全部放在大于等于x的节点左边，且小于部分和大于等于部分的节点顺序必须是初始顺序。看到这道题后，就想到创建两个链表，一个链表存储小于x的节点，一个链表存储大于等于x的节点，然后依次遍历原始的链表，遍历完之后将大于等于x的节点接在小于x的节点后面即可。

# 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:

        # 定义两个链表，一个中全是小于x的，一个中全是大于等于x的
        minx = ListNode(0)
        p1 = minx
        maxx = ListNode(0)
        p2 = maxx

        p = head
        while p is not None:
            tmp = ListNode(p.val)
            if p.val < x:
                p1.next = tmp
                p1 = tmp
            else:
                p2.next = tmp
                p2 = tmp
            p = p.next
            
        p1.next = maxx.next
        return minx.next

今天看了一天综艺，罪过罪过，赶紧做一道题弥补一下。而且机智如我居然跳题了。过两天再看跳的那道困难吧。