LeetCode-86. Partition List

题目描述

Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

Example:

Input: head = 1->4->3->2->5->2, x = 3
Output: 1->2->2->4->3->5

• 原题链接
• 题目给定的输入为一个链表的头节点和一个整型的数字$x$,以$x$为分界符，对原链表节点顺序重新划分，小于$x$的在左侧，大于的在右侧，但是节点之间的相对位置不变。

解法1

思路

• 建立两个临时头节点，一个用作小于$x$的节点的链接，一个用于大于$x$的链接
• 分别建立两个工作指针，完成上述节点的链接工作，工作指针始终指向当前临时头节点所在链表的最后一个节点，临时头节点的地址保持不变。
• 遍历原链表，完成上述操作
• 遍历完成后，连接上上述部分

代码

static ListNode partition(ListNode head, int x){
        //用于小于x节点的临时头节点
        ListNode head_less = new ListNode(0);
        //工作指针
        ListNode ptr_less = head_less;
        //用于大于x节点的临时头节点
        ListNode head_more = new ListNode(0);
        ListNode ptr_more = head_more;

        while (head!=null){//遍历原链表
            if (head.val < x){
                //如果小于x，添加到head_less后面
                //添加过程由ptr_less来完成
                ptr_less.next = head;
                ptr_less = head;
            }else {
                //如果大于x，添加到head_more后面
                ptr_more.next = head;
                ptr_more = head;
            }
            head = head.next;//遍历原链表
        }

        //完成遍历后，连接两部分
        ptr_more.next = null;
        ptr_less.next = head_more.next;
        //注意返回新链表的第一个节点，而不是临时头节点
        return head_less.next;

    }

细节

• 注意工作指针的应用，在链表的题目里非常常见
• 注意完成遍历后的尾节点的下一个节点应该置空
  • ptr_more.next = null;
• 注意返回的应是新链表的第一个节点位置
  • return head_less.next;