leetcode 86. 分隔链表

'''
leetcode 86. 分隔链表
给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。

算法：
通过在单向链表的特定位置上插入节点和删除节点实现
设定初始指针从前向后遍历，首先找到第一个数值大于或者等于x的节点的直接前驱节点，记作key_node
这间接保证了在key_node节点之前的节点的数值都是小于x的，故而只需要对key_node节点之后的节点进行操作即可
找到一个小于x的节点，将该节点从原位置上删除，并插入到key_node的后面，而key_node指向当前被插入的节点
在此过程中需要4个指针记录节点的位置：
key_node节点的前驱，key_node节点，和curr节点(即当前扫描到的节点)以及curr节点的前驱节点curr_before
'''
# 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:
        if head is None or head.next is None:
            return head
        # 如果链表的头结点就是大于或者等于x的节点，需要向头结点之前插入节点，为此在头节点之前引入哨兵节点
        new_head=ListNode(-float('inf'))
        new_head.next=head
        curr=new_head
        prev=new_head
        key=new_head
        while(prev.next is not None):
            if prev.next.val<x:
                # print('prev.next.val',prev.next.val)
                prev=prev.next
            else:
                break
        # print(prev.val)
        if prev.next is None:
            # 说明整个链表中节点的数值都是小于x的，故而直接返回链表的头结点即可
            return head
        else:
            # 说明找到了prev.next节点就是单向链表中第一个节点数值大于x的节点
            key=prev.next
            if key.next is None:
                # 说明当前的单向链表中只有最后一个元素是大于或者等于x的节点，在此之前的所有节点的数值都小于x
                # 此时的单向链表自然就是前后分隔好的两个部分，直接返回头结点即可
                return head
            curr_before=key
            curr=key.next
            while(curr is not None):
                if curr.val<x:
                    # 首先根据cur的数值创建新节点，并将新节点插入到prev和key节点之间
                    # 然后删除curr节点，即将curr的前驱指向curr的后继
                    # new_node=ListNode(curr.val)
                    curr_before.next = curr.next
                    curr.next=key
                    prev.next=curr
                    prev=curr

                    curr=curr_before.next

                else:
                    curr_before=curr
                    curr=curr.next
            return new_head.next

if __name__=="__main__":
    new_head=ListNode(0)
    curr=new_head
    for i in [1,4,3,2,5,2]:
        new_node=ListNode(i)
        curr.next=new_node
        curr=curr.next

    output=Solution().partition(new_head.next,3)
    while(output):
        print('here',output.val)
        output=output.next