LeetCode 86分隔链表 Python题解

最新推荐文章于 2023-04-29 16:45:29 发布

热衷Python

最新推荐文章于 2023-04-29 16:45:29 发布

阅读量111

点赞数 1

文章标签：链表 leetcode 数据结构 python

本文链接：https://blog.csdn.net/weixin_45450206/article/details/130189585

版权

该问题讨论如何在保持节点相对位置不变的情况下，将链表分成两部分，一部分包含所有小于给定值x的节点，另一部分包含大于或等于x的节点。解决方案是创建两个新的链表，分别存储这两个部分的节点，最后将它们连接起来。关键在于不直接在原链表上操作，因为链表节点没有前驱节点的引用。

摘要由CSDN通过智能技术生成

分隔链表链接
最近没有更新是换工作，换房子去了，有点忙，抱歉

# 分隔链表
"""
给你一个链表的头节点head和一个特定值x ，请你对链表进行分隔，使得所有小于x的节点都出现在大于或等于x的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
输入：head = [1,4,3,2,5,2], x = 3
输出：[1,2,2,4,3,5]
"""

"""
    这道题注意一下相对位置比较重要，不要想着在一个链表中去移动，因为没法移动
    原因就是这个链表对象没有前置节点对象的保存，如果有就可以一个链表移动
    所以说就是两个链表:一个保存比x小的 一个保存大于等于x的 只要保存必然是相对位置
    最后连接 因为两个链表后面都有0 做了一些处理 不这样写也可以。
"""


# Definition for singly-linked list.
class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next


class Solution:
    def partition(self, head, x):
        if head is None or head.next is None:
            return head
        t1, t2 = ListNode(), ListNode()  # head = [1,4,3,2,5,2]
        res1, res2 = ListNode(), ListNode()
        res1.next, res2.next = t1, t2
        while head:
            if head.val >= x:
                t1.val = head.val
                t1.next = ListNode()
                t1 = t1.next
            else:
                t2.val = head.val
                t2.next = ListNode()
                t2 = t2.next
            head = head.next
        t2.val = res1.next.val
        t2.next = res1.next.next
        temp = res2.next
        while temp:
            if temp.next.next is None:
                temp.next = None
                break
            temp = temp.next

        return res2.next