链表(把链表数值分类,时间复杂度为O(N),空间复杂度为O(1))

给定一个链表和一个target,把链表分为小于target部分,等于target,大于target三个部分。代码如下:

struct Node
{
    int val;
    Node *next;
};
Node* Partition(Node* head,int pivot)
{
    Node *sH=NULL;//small head
    Node *sT=NULL;//small tail
    Node *eH=NULL;
    Node *eT=NULL;
    Node *mH=NULL;
    Node *mT=NULL;
    while(head!=NULL)
    {
        Node *next=head->next;
        head->next=NULL;
        if(head->val<pivot)
        {
            if(sH==NULL)
            {
                sH=head;
                sT=head;
            }
            else {
            sT->next=head;
            sT=head;
            }

        }
        else if(head->val==pivot)
        {
            if(eH==NULL)
            {
                eH=head;
                eT=head;
            }
            else{
                eT->next=head;
                eT=head;
            }
        }
        else
        {
            if(mH==NULL)
            {
                mH=head;
                mT=head;
            }
            else{
                mT->next=head;
                mT=head;
        }
        }
        head=next;
    }
    if(sT!=NULL)
    {
        sT->next=eH;
        eT=eT==NULL? sT:eT;
    }
    if(eT!=NULL)
    {
        eT->next=mH;
    }
    return sH !=NULL? sH:(eH!=NULL?eH:mH);

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值