[牛客OJ]链表分割Java

Myth丶恋晨 2023-06-27 03:47 104阅读 0赞

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();
        }
    }