题目:
分隔链表:给定一个链表和一个特定值 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