力扣 leetcode 86. 分隔链表

Topic:
给你一个链表和一个特定值 x ,请你对链表进行分隔,使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

Example:

输入:head = 1->4->3->2->5->2, x = 3
输出:1->2->2->4->3->5

思路:
整体的思路就是将链表按原序拆分为小的和大的两个链表,最后合并

首先先设置两个哑节点(便于边界的判断)
之后遍历整个链表
若head的val小于x,则small的下一位变为head
small指针转换为下一位(此时head的val)
同理
若head的val大于或等于于x,则big的下一位变为head
big指针指向下一位(此时head的val)
这样两个链表已经按照原来的顺序
分别将小的保存为了small
大的保存为了big
遍历结束后
将 big 的 next 指针清空(防止指向原来初始链表的节点)
同时将 small 的 next 指向L2的next 的节点(big链表真正的头节点)
最后返回L1的next指针

Code:

class Solution:
    def partition(self, head: ListNode, x: int) -> ListNode:
        L1 = small = ListNode(0)
        L2 =  big = ListNode(0)
        
        while head:
            if head.val < x:
                small.next = head
                small = small.next
            else:
                big.next = head
                big = big.next
            
            head = head.next

        big.next = None
        small.next = L2.next
        
        return L1.next

Result:

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_50791900/article/details/112131156