链表分割

来源：互联网发布：面板数据计量经济学编辑：程序博客网时间：2024/06/07 02:11

题目描述

编写代码，以给定值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;    }}

0 0