将所有小于x的结点排在其余结点之前，且不能改变原来的数据顺序

Posted 2022-12-11 你快看看我

题目描述：现有一链表的头指针 ListNode* pHead，给一定值x，编写一段代码将所有小于x的结点排在其余结点之前，且不能改变原来的数据顺序，返回重新排列后的链表的头指针。

输入：2 4 9 3 2 1 6，x=5
输出：2 4 9 3 2 1 6

思路：遍历链表，依次拿链表的每个节点的val和x对比，开辟两个新的链表，将小于x的节点存入一个链表，将大于或者等于的节点存入另一个链表，最后将大的链表的头节点接入小的链表的尾部。
代码实现如下：

struct ListNode* partition(struct ListNode* pHead, int x)
    
    //开辟两个新的链表
        struct ListNode* lessHead;
        struct ListNode* lessTail;
        struct ListNode* greaterHead;
        struct ListNode* greaterTail;
        lessHead = lessTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        greaterHead = greaterTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        lessTail->next = NULL;
        greaterTail->next = NULL;

        struct ListNode* cur = pHead;
        while (cur)//比较val和x
        
            if (cur->val < x)
            
                lessTail->next = cur;
                lessTail = lessTail->next;
            
            else
            
                greaterTail->next = cur;
                greaterTail = greaterTail->next;
            
            cur = cur->next;
        
        
        //链接两个链表
        lessTail->next = greaterHead->next;

        pHead = lessHead->next;
        free(lessHead);
        lessHead = NULL;
        free(greaterHead);
        greaterHead = NULL;
        return pHead;
    

但是其实上面的代码有一些案例可能无法实现，因为有些输入可能会造成链表成了一个圈。
例如;
输入：2 4 9 3 2 1 6 3，x=5
输出：2 4 9 3 3 2 1 6
如果输入这个案例，编译器就会报错，提示你内存不足，但是你只开辟了两个新的链表，怎么会内存过多呢？
这个案例和上一个案例相比区别在于输入最后多了一个3
因为在greater这个链表的最后一个节点是6，但是在原输入俩表中6->3，6的结点后面存入的是3，当while循环完成以后，6->next并没有置为NULL，所以这个6->next依然指向3，但是3是lessTail,将lessTail->next=greater->next时，会形成一个循环。

所以应该在while循环结束以后加一句
greaterTail->next=NULL;
更新后的代码如下：

struct ListNode* partition(struct ListNode* pHead, int x)
    
        struct ListNode* lessHead;
        struct ListNode* lessTail;
        struct ListNode* greaterHead;
        struct ListNode* greaterTail;
        lessHead = lessTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        greaterHead = greaterTail = (struct ListNode*)malloc(sizeof(struct ListNode));
        lessTail->next = NULL;
        greaterTail->next = NULL;

        struct ListNode* cur = pHead;
        while (cur)
        
            if (cur->val < x)
            
                lessTail->next = cur;
                lessTail = lessTail->next;
            
            else
            
                greaterTail->next = cur;
                greaterTail = greaterTail->next;
            
            cur = cur->next;
        

        greaterTail->next = NULL;
        lessTail->next = greaterHead->next;

        pHead = lessHead->next;
        free(lessHead);
        lessHead = NULL;
        free(greaterHead);
        greaterHead = NULL;
        return pHead;