作业8.96：链表划分

题目：链表划分

样例：

给定1->4->3->2->5->2->null，并且x=3。

返回1->2->2->4->3->5->null。

思路：设置两个新链表small和big，遍历原始链表，小于x的元素放在small里，大于x的元素放在big里，然后big连在small后面返回。

代码：

class Solution {
public:
    /**
     * @param head: The first node of linked list.
     * @param x: an integer
     * @return: a ListNode 
     */
        ListNode *partition(ListNode *head, int x) {  
        // write your code here  
        if(head==NULL)   
            return NULL;  
        ListNode *small=new ListNode(0);  
        ListNode *big=new ListNode(0);  
        ListNode *p=small;  
        ListNode *q=big;  
        while(head!=NULL)  
        {  
            if(head->val<x)  
            {  
                p->next=head;  
                p=p->next;  
            }  
            else  
            {  
                q->next=head;  
                q=q->next;  
            }  
            head=head->next;  
        }  
        q->next=NULL;
        p->next=big->next;  
        return small->next;  
    }  
};  

感想：一开始没有令q->next=NULL，运行总是显示内存不够，空间复杂度超标，调了好多地方才找到。