public class ListSplit {
private static ListNode partition(ListNode pHead, int x) {
// write code here
if(pHead==null){
return null;
}
//创建x结点作为分割前半部分和后半部分的中间结点
ListNode nodex=new ListNode(x);
//创建newHead结点方便第一个小于x值的插入
ListNode newHead=new ListNode(0);
newHead.next=nodex;
//创建before结点,在迭代过程中始终保持before.next=nodex
//从而保证小于x值的结点可以插入到nodex结点之前
ListNode before=newHead;
//创建after结点,在迭代过程中始终保持after结点是最后一个结点
//从而保证大于等于x值的结点可以插入链表的最后位置
ListNode after=nodex;
ListNode walkNode=pHead;
boolean isexist=false;
while(walkNode!=null){
//如果当前节点小于x,复制结点并将其插入到xnode的前一个结点,然后移动before指针
if(walkNode.val<x){
ListNode node=new ListNode(walkNode.val);
before.next=node;
node.next=nodex;
before=node;
}
//如果当前节点大于x,复制结点并将其插入到链表最后一个结点,然后移动after指针
else if(walkNode.val>=x){
ListNode node=new ListNode(walkNode.val);
after.next=node;
after=node;
}
walkNode=walkNode.next;
}
//忽略自建的x结点nodex和头结点newHead;
//因为x结点不一定存在于原链表,所以此处要将分开的前后部分相连
before.next=nodex.next;
return newHead.next;
}
public static void main(String []args){
ListNode head = new ListNode(3);
ListNode node2 = new ListNode(3);
ListNode node3 = new ListNode(3);
ListNode node4 = new ListNode(3);
head.next = node2;
node2.next = node3;
node3.next = node4;
ListSplit split =new ListSplit();
ListNode returnHead=split.partition(head,3);
while (returnHead != null)
{
System.out.println(returnHead.val );
returnHead = returnHead.next;
}
System.out.println();
}
}
[牛客OJ]链表分割Java
猜你喜欢
转载自blog.csdn.net/qq_44057443/article/details/103812606
今日推荐
周排行