题目描述:
编写代码,以给定值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;
}
}