刷leetcode-day32

一、题目:

86.分隔链表

给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

二、思路:

首先用指针p找出第一个值大于等于x的结点,把p指向该结点的前一个结点。用指针q遍历p之后的结点,如果结点的值小于x的值,则把该结点放到p结点的后面。

三、代码(C++):

ListNode* partition(ListNode* head, int x) {
        ListNode *h=new ListNode(0);      //新建一个头结点
        h->next=head;
        head=h;
        ListNode *p=h;
        while(p->next)                              //寻找第一个大于等于x的结点,把p指向该结点之前
        {
            if(x<=p->next->val)
                break;
            else
                p=p->next;
        }
        ListNode *q=p;
        while(q->next)                              //把p结点之后的每一个值小于x的结点放到p结点的后面
        {
            if(q->next->val<x)
            {
                ListNode *f=q->next;
                q->next=q->next->next;
                f->next=p->next;
                p->next=f;
                p=p->next;
            }
            else
                q=q->next;
        }
        return head->next;
    }

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务