86. 分隔链表(中等系列)

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

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

示例 1:
86. 分隔链表(中等系列)_第1张图片
输入:head = [1,4,3,2,5,2], x = 3
输出:[1,2,2,4,3,5]
示例 2:

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

提示:

链表中节点的数目在范围 [0, 200] 内
-100 <= Node.val <= 100
-200 <= x <= 200

public ListNode partition(ListNode head, int x) {
    // 初始化左半部分链表的头和尾节点
    ListNode leftHead = null, leftTail = null;
    // 初始化右半部分链表的头和尾节点
    ListNode rightHead = null, rightTail = null;
    // 用于保存下一个节点的指针
    ListNode next = null;
    
    // 遍历链表节点
    while (head != null) {
        // 保存当前节点的下一个节点
        next = head.next;
        // 断开当前节点与下一个节点的连接
        head.next = null;
        
        // 判断节点值与 x 的大小关系
        if (head.val < x) {
            if (leftHead == null) {
                leftHead = head;
            } else {
                leftTail.next = head;
            }
            leftTail = head;
        } else {
            if (rightHead == null) {
                rightHead = head;
            } else {
                rightTail.next = head;
            }
            rightTail = head;
        }
        
        // 移动到下一个节点
        head = next;
    }
    
    // 如果左半部分链表为空,直接返回右半部分链表
    if (leftHead == null) {
        return rightHead;
    }
    
    // 将左半部分链表的尾节点与右半部分链表的头节点连接起来
    leftTail.next = rightHead;
    // 返回整合后的链表
    return leftHead;
}

你可能感兴趣的:(Leetcode,链表,数据结构)