牛客网CM11 链表分割

LuckyRich1

已于 2023-04-28 22:54:16 修改

阅读量185

点赞数 1

分类专栏：牛客文章标签：链表 c语言数据结构

于 2023-04-28 11:55:29 首次发布

本文链接：https://blog.csdn.net/fight_p/article/details/130420928

版权

牛客专栏收录该内容

10 篇文章 0 订阅

订阅专栏

该问题要求将一个链表按值分割，所有小于x的节点位于其他节点之前，且保持原有顺序。提供了两种解法：1)使用两个额外的头结点，分别构建小于x的节点链表和大于等于x的节点链表，然后连接；2)不使用额外头结点，直接在原链表上操作，动态维护两个链表，最后处理空链表情况。

摘要由CSDN通过智能技术生成

描述

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

示例

在这里插入图片描述

解题思路以及代码

解法1

申请两个头结点，小的放在minhead，大的放在maxhead，再把两个链表连接起来。
在这里插入图片描述

class Partition {
public:
    ListNode* partition(ListNode* pHead, int x) {
        // write code here
        struct ListNode* minhead,*mintail,*maxhead,*maxtail;
        maxtail=maxhead=(struct ListNode*)malloc(sizeof(struct ListNode));
        mintail=minhead=(struct ListNode*)malloc(sizeof(struct ListNode));
		minhead=maxhead=NULL;
        struct ListNode* cur=pHead;
        while(cur)
        {
            if(cur->val < x)
            {
                mintail->next=cur;
                mintail=mintail->next;
            }
            else
            {
                maxtail->next=cur;
                maxtail=maxtail->next;
            }
            cur=cur->next;
        }
        mintail->next=maxhead->next;
        maxtail->next=NULL;
        struct ListNode* head=minhead->next;
        free(minhead);
        free(maxhead);
        return head;
    }
};

解法2

不申请头结点，小的放在minhead，大的放在maxhead,然后链接起来，这种做法要注意，minhead和maxhea链表为空的情况，要做近一步处理
在这里插入图片描述

class Partition {
public:
    ListNode* partition(ListNode* pHead, int x) {
        // write code here
        struct ListNode* minhead,*mintail,*maxhead,*maxtail;
        minhead=mintail=NULL;
        maxhead=maxtail=NULL;
        struct ListNode* cur=pHead;
        while(cur)
        {
            if(cur->val < x)
            {
                if(mintail == NULL)
                {
                    minhead=mintail=cur;
                }
                else 
                {
                    mintail->next=cur;
                    mintail=mintail->next;
                }

            }
            else
            {
                if(maxtail == NULL)
                {
                    maxhead=maxtail=cur;
                }
                else 
                {
                    maxtail->next=cur;
                    maxtail=maxtail->next;
                }

            }
            cur=cur->next;
        }
        //maxhead或minhead链表为空
        if (maxtail == NULL)
        {
            return minhead;
        }
        else if (mintail == NULL)
        {
            return maxhead;
        }
        else
        {
            mintail->next = maxhead;
            maxtail->next = NULL;
            return minhead;
        }
        
    }
};