精通链表：LeetCode 第 86 题“分隔链表”详解

2023-01-08 08:48:08

踏上 LeetCode 挑战之旅，分隔链表登巅峰

作为一名矢志不渝的代码行者，每日算法题已成为我的必修课。LeetCode 第 86 题“分隔链表”赫然出现在我的面前，宛如一道中等难度的珠穆朗玛峰，激起了我熊熊的征服欲。在这篇博文中，我们将携手踏上征服这座算法高峰的征程，共同领略分隔链表的奥秘，共攀算法的巅峰。

剖析题意，撥开迷霧見真知

首先，让我们來仔细审视这道题目的题意，拨开迷雾，洞悉其本质。题目给定一个链表和一个整数 x，要求将链表分成两部分：一部分包含所有小于或等于 x 的节点，另一部分包含所有大于 x 的节点。两部分的顺序可以任意。

解题思路，拨云见日柳暗花明

在吃透题意的基础上，我们可以开始思考解决问题的方案。这里介绍一种巧妙的解题思路：

初始化两个链表： 一个用于存储小于或等于 x 的节点，另一个用于存储大于 x 的节点。
遍历原链表： 逐个遍历原链表的每个节点，并将每个节点与 x 进行比较。
将节点分配到相应链表： 如果当前节点小于或等于 x，则将其添加到小于或等于 x 的链表中；否则，将其添加到大于 x 的链表中。
连接两个链表： 最后，将两个链表连接起来，就得到了满足题目要求的新链表。

代码示例，一览无余步步为营

为了更好地理解解题思路，我们来看一下示例代码：

def partition(head, x):
    # 初始化两个链表
    less_or_equal_head = ListNode(0)
    greater_head = ListNode(0)

    # 遍历原链表
    current = head
    while current:
        # 将节点分配到相应链表
        if current.val <= x:
            less_or_equal_head.next = current
        else:
            greater_head.next = current

        # 更新 current 指针
        current = current.next

    # 连接两个链表
    less_or_equal_head.next = greater_head.next

    # 返回小于或等于 x 的链表的头节点
    return less_or_equal_head.next