Leetcode练习(Python):链表类:第86题:分隔链表:给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 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:
        if not head:
            return None
        linkedList1 = ListNode(0)
        linkedList2 = ListNode(0)
        myNode1 = linkedList1
        myNode2 = linkedList2
        myNode = head
        while myNode:
            tmp_node = myNode.next
            if myNode.val < x:
                myNode.next = None
                myNode1.next = myNode
                myNode1 = myNode1.next
            else:
                myNode.next = None
                myNode2.next = myNode
                myNode2 = myNode2.next
            myNode = tmp_node
        myNode1.next = linkedList2.next
        return linkedList1.next

猜你喜欢

转载自www.cnblogs.com/zhuozige/p/12819135.html