按照大小分割链表后合并链表

编写程序以 x 为基准分割链表,使得所有小于 x 的节点排在大于或等于 x 的节点之前。如果链表中包含 x,x 只需出现在小于 x 的元素之后(如下所示)。分割元素 x 只需处于“右半部分”即可,其不需要被置于左右两部分之间。

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

       直观来说我们只需维护两个链表 small\textit{small}small 和 large\textit{large}large 即可,small\textit{small}small 链表按顺序存储所有小于 xxx 的节点,large\textit{large}large 链表按顺序存储所有大于等于 xxx 的节点。遍历完原链表后,我们只要将 small\textit{small}small 链表尾节点指向 large\textit{large}large 链表的头节点即能完成对链表的分割。
为了实现上述思路,我们设 smallHead\textit{smallHead}smallHead 和 largeHead\textit{largeHead}largeHead 分别为两个链表的哑节点,即它们的 next\textit{next}next 指针指向链表的头节点,这样做的目的是为了更方便地处理头节点为空的边界条件。同时设 small\textit{small}small 和 large\textit{large}large 节点指向当前链表的末尾节点。开始时 smallHead=small,largeHead=large\textit{smallHead}=\textit{small},\textit{largeHead}=\textit{large}smallHead=small,largeHead=large。随后,从前往后遍历链表,判断当前链表的节点值是否小于 xxx,如果小于就将 small\textit{small}small 的 next\textit{next}next 指针指向该节点,否则将 large\textit{large}large 的 next\textit{next}next 指针指向该节点。
遍历结束后,我们将 large\textit{large}large 的 next\textit{next}next 指针置空,这是因为当前节点复用的是原链表的节点,而其 next\textit{next}next 指针可能指向一个小于 xxx 的节点,我们需要切断这个引用。同时将 small\textit{small}small 的 next\textit{next}next 指针指向 largeHead\textit{largeHead}largeHead 的 next\textit{next}next 指针指向的节点,即真正意义上的 large\textit{large}large 链表的头节点。最后返回 smallHead\textit{smallHead}smallHead 的 next\textit{next}next 指针即为我们要求的答案。(leetcode官方回答,自附步骤图如下:)

class Solution {
    public ListNode partition(ListNode head, int x) {
        ListNode small = new ListNode(0);
        ListNode smallHead = small;
        ListNode large = new ListNode(0);
        ListNode largeHead = large;
        while (head != null) {
            if (head.val < x) {
                small.next = head;
                small = small.next;
            } else {
                large.next = head;
                large = large.next;
            }
            head = head.next;
        }
        large.next = null;
        small.next = largeHead.next;
        return smallHead.next;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_44624536/article/details/115365298
今日推荐