链表面试-链表分割（常考题）

Posted 2021-09-03 eaglewgs

题目描述：

编写代码，以给定值x为基准将链表分割成两部分，所有小于x的节点排在大于或等于x的节点之前，给定一个链表的头指针ListNode* pHead,请返回重新排列后的链表的头指针。注意：分割以后保持原来的数据顺序不变。

示例描述：

x=12,原链表如下：

两个线段存放节点过程：

分割后的链表如下：

个人理解：

• 本题用两个线段来存放比x小的节点和大于等于x的节点，刚开始的时候两个线段的头尾bs,be;as,ae;先置为null
• 定义一个cur,在原来的链表中进行遍历
• 如果cur.val<x,那么将该节点放到第一个线段，否则放到第二个线段，如果第一个线段没有数据，那么直接返回as;如果第一个线段有数据，那么将be和as进行拼接，be.next = as。
• 当cur为null的时候，原来的单链表就遍历完了

题解实现：

import java.util.*;

/*
public class ListNode {
    int val;
    ListNode next = null;

    ListNode(int val) {
        this.val = val;
    }
}*/
public class Partition {
    public ListNode partition(ListNode pHead, int x) {
        ListNode bs = null;//将两段的首尾先置为null
        ListNode be = null;
        ListNode as = null;
        ListNode ae = null;
        ListNode cur = pHead;
        while(cur!= null){
            if(cur.val < x){//再分为第一次插入以及不是第一次插入
                if(bs==null){//第一次插入
                    bs = cur;
                    be = cur;
                }else{//不是第一次插入,就进行尾插法
                    be.next = cur;
                    be = be.next;
                }
            }else {//同样的也再分为第一次插入以及不是第一次插入
                if(as == null){//第一次插入
                    as = cur;
                    ae = cur;
                }else{//不是第一次插入
                    ae.next = cur;
                    ae = ae.next;
                }
            }
            cur = cur.next;
        }
        //1.判断bs是否为null,如果bs==null，则返回as
        if(bs == null){
            return as;
        }
        //2.如果bs不为null，需要进行拼接
        be.next = as;
        //3.如果ae不为null，那么ae的next需要置为null,此时返回bs
        if(ae!=null){
            ae.next = null;
        }
        return bs;
    }
}