CM11 链表分割题解

news2025/2/25 11:16:13

题目描述：

链表分割_牛客题霸_牛客网 (nowcoder.com)

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

题解思路：

分别创建两个新链表 less、large，并用一个指针current指向原始链表的头节点用于遍历；这里位于后续尾插可以使用带哨兵位头节点的链表和尾指针指向两链表的尾端；
比较current指向节点的值与给定值x的大小关系：如果current节点的值小于x，将current的值作为新节点的值尾插到 less 头节点的后面，移动current；如果current节点的值大于、等于x，将current的值作为新节点的值尾插到 large 头节点的后面，移动 current；
重复步骤二，直到current遍历所有节点，指向NULL；
连接 less 和 large ：用 less 的尾指针的指针域指向 large 头节点的下一个；
返回 less 头结点的下一个。

代码：

ListNode* partition(ListNode* pHead, int x) 
    {
        struct ListNode* lessHead = (ListNode*)malloc(sizeof(ListNode));
        struct ListNode* largeHead = (ListNode*)malloc(sizeof(ListNode));
        lessHead->next = NULL;
        largeHead->next = NULL;
        struct ListNode* lessTail = lessHead;
        struct ListNode* largeTail = largeHead;
        struct ListNode* current = pHead;
        while(current)
        {
            // 比较
            if(current->val < x)
            {
                // 尾插
                struct ListNode* newNode = (ListNode*)malloc(sizeof(ListNode));
                newNode->next = NULL;
                newNode->val = current->val;
                lessTail->next = newNode;
                lessTail = newNode;
            }
            else 
            {
                struct ListNode* newNode = (ListNode*)malloc(sizeof(ListNode));
                newNode->next = NULL;
                newNode->val = current->val;
                largeTail->next = newNode;
                largeTail = newNode;
            }
            current = current->next;
        }
        // 连接
        lessTail->next = largeHead->next;
        return lessHead->next;
    }

注：这里尾插的的时候可以不进行创建新的节点，可以直接用原链表的节点进行连接即可。

ListNode* partition(ListNode* pHead, int x)
     {
       struct ListNode* lessHead, *lessTail  , *largeHead, *largeTail, *current;
       lessHead = lessTail = (struct ListNode*)malloc(sizeof(struct ListNode));
    largeHead = largeTail = (struct ListNode*)malloc(sizeof(struct  ListNode));
    lessTail->next = NULL;
    largeTail->next = NULL;
    current = phead;
 
    while (current != NULL)
    {
        if (current->val < x)
        {
                lessTail->next = current ;
                lessTail = current ;
        }
        else
        {
                largeTail->next = current ;
                largeTail = current ;
        }
        current = current->next;
 
    }
        lessTail->next = largeHead->next;
        ListNode* temp = lessHead->next;
        free(lessHead);
        free(largeHead);
        return temp;
}