careercup-链表 2.4

#include<iostream>
#include<new>
using namespace std;

struct ListNode
{
    int val;
    ListNode *next;
    ListNode(int x):val(x),next(NULL) {}
};

void createList(ListNode *&L)
{
    int arr[10]= {1,2,3,2,5,6,7,3,9,1};
    int i;
    ListNode *p=NULL;
    for(i=0; i<10; i++)
    {
        ListNode *tmp=new ListNode(arr[i]);
        if(L==NULL)
        {
            L=tmp;
            p=tmp;
        }
        else
        {
            p->next=tmp;
            p=tmp;
        }
    }
}

void splitX(ListNode *L,int x)
{
    if(L==NULL)
        return;
    ListNode *pre=L;
    ListNode *p=L;
    ListNode *L1=NULL;
    ListNode *q=NULL;
    while(p)
    {
        if(p->val<x)
        {
            pre=p;
            p=p->next;
        }
        else
        {
            ListNode *tmp=p;
            p=tmp->next;
            pre->next=p;
            tmp->next=NULL;
            if(L1==NULL)
            {
                L1=tmp;
                q=tmp;
            }
            else
            {
                q->next=tmp;
                q=tmp;
            }
        }
    }
    pre->next=L1;
}

int main()
{
    ListNode *head=NULL;
    createList(head);
    ListNode *p=head;
    while(p)
    {
        cout<<p->val<<" ";
        p=p->next;
    }
    cout<<endl;
    splitX(head,5);
    p=head;
    while(p)
    {
        cout<<p->val<<" ";
        p=p->next;
    }
    cout<<endl;
}