链表分割

题目描述

编写代码，以给定值x为基准将链表分割成两部分，所有小于x的结点排在大于或等于x的结点之前
给定一个链表的头指针 ListNode* pHead，请返回重新排列后的链表的头指针。

Thinking

如图所示我们建立两个链表一个专门存放小于x的节点，另一个专门存放大于x的节点，访问原链表如果current节点大于x放到qe后面否则放在pe后面最后将两链表合并，返回ps引用即可。（如果ps==nll即原链表数据均大于必须返回qs）

Code

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 ps=null;
        ListNode pe=null;
        ListNode qs=null;
        ListNode qe=null;
        ListNode current=pHead;
        while(current!=null){
            ListNode temp= current.next;
            current.next=null;
            if(current.val<x){
                if(ps==null){

                    ps=current;
                    pe=current;
                }
                else{
                    pe.next=current;
                    pe=current;
                }
            }
            else{
                 if(qs==null){

                    qs=current;
                    qe=current;
                }
                else{
                    qe.next=current;
                    qe=current;
                }
            }
            current=temp;

        }
        if(ps==null){
            return qs;
        }
        pe.next=qs;
        return ps;
    }
}