力扣每日一题：86.分隔链表创建大小链表与寻找第一个链表头两种思路

最新推荐文章于 2022-11-09 19:53:18 发布

清风Python

最新推荐文章于 2022-11-09 19:53:18 发布

阅读量134

点赞数

分类专栏：算法之美文章标签：链表 python 算法 leetcode java

本文链接：https://blog.csdn.net/BreezePython/article/details/117772875

版权

算法之美专栏收录该内容

97 篇文章 5 订阅

订阅专栏

86.分隔链表

https://leetcode-cn.com/problems/partition-list/solution/86fen-ge-lian-biao-chuang-jian-da-xiao-l-gy5y/

难度：中等

题目：

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

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

示例：

示例 1：
输入：head = [1,4,3,2,5,2], x = 3
输出：[1,2,2,4,3,5]
示例 2：

输入：head = [2,1], x = 2
输出：[1,2]

分析

这道题最简单的解题方式，就是我们创建两个小链表和大链表。
然后循环head，当head.val小于x时，追加到小链表，否则添加到大链表
最终将小链表与大链表合并。然后将大链表的链尾置为None。

当然，为了熟悉链表操作，写了一种寻找第一个大链表头的解法。
创建一个空链表，然后循环判断head，当head.val小于x，则直接拼接ret.next = head
当找到第一个大于等于x的val时，设置为刚才我们创建的大链表头，之后进行对比插入。
这个解法是为了进一步熟悉链表操作，自娱自乐...

解题1-双链表合并：

class Solution:
    def partition(self, head, x):
        if not head:
            return head
        small = ListNode()
        sm = small
        large = ListNode()
        lg = large
        while head:
            if head.val < x:
                sm.next = head
                sm = sm.next
            else:
                lg.next = head
                lg = lg.next
            head = head.next
        lg.next = None
        sm.next = large.next
        return small.next

解题2-寻找第一个large链表头：

class Solution:
    def partition(self, head, x):
        tmp = ret = ListNode()
        li = head
        ins = None
        while li:
            if li.val < x:
                tmp.next = li
                tmp = tmp.next
                li = li.next
            else:
                ins = li
                break
        while li and li.next:
            if li.next.val < x:
                tmp.next = li.next
                tmp = li.next
                li.next = li.next.next
            else:
                li = li.next
        tmp.next = ins
        return ret.next

欢迎关注我的公众号: 清风Python，带你每日学习Python算法刷题的同时，了解更多python小知识。

我的个人博客：https://qingfengpython.cn

力扣解题合集：https://github.com/BreezePython/AlgorithmMarkdown