牛客网-CM11-链表分割

题目描述

现有一链表的头指针 ListNode* pHead，给一定值x，编写一段代码将所有小于x的结点排在其余结点之前，且不能改变原来的数据顺序，返回重新排列后的链表的头指针。

---

思路

创建两个链表，分别存储大于x和小于x的结点，然后将两个结点连接在一起
注意：创建的两个链表头结点是没有意义的（不使用）

---

代码

public class Partition {
    public ListNode partition(ListNode pHead, int x) {
        // write code here
        ListNode headA = new ListNode(-1);
        ListNode headB = new ListNode(-1);
       
        ListNode lastA = headA;
        ListNode lastB = headB;
         
        for(ListNode cur = pHead; cur != null; cur = cur.next){
            if(cur.val < x){
                lastA.next = cur;
                lastA = cur;
            }else{
                lastB.next = cur;
                lastB = cur;
            }
        }
         
        lastA.next = headB.next;
        lastB.next = null;//这一步必须有，下面解释
        return headA.next;  
    }
}

---

解释代码

1、创建两个链表，头结点不使用，并且定义链表的尾结点

2、遍历原始链表，将所有小于x的结点尾插到链表A中，将所有大于x的结点尾插到链表B中。

3、连接两个链表
将A链表的尾结点与B链表头结点的下一个结点连接，并将链表B的尾结点指向空

将链表B指向null这一步是必不可少的，因为链表lastB是从原链表中切割出的，后边可能还有连接的结点，所以需要置空，不置空有可能会陷入死循环。