Leetcode 分开链表

给定一个链表和一个值x,利用x把链表进行划分,比x小的都排列在比x大或者等于x的后面。链表原来元素的相对位置不变。

For example,
Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.

解答:想法是从前向后遍历,遇到第一个比x大的元素的时候停下,记录位置p1,同时始终有一个pre1记录p1之前的节点。这个时候从p1开始向后找,找到第一个比x小的元素节点,记录为p2,同时保持了p2的前驱节点。

再将p2插入到p1之前,同时对相应指针进行更新,进行下一轮的循环。

ListNode *partition(ListNode *head, int x) {
        if(head == NULL || head->next == NULL) return head;
        ListNode *p1 = head, *p2 = head,*dummy,*pre1,*pre2;
        dummy = new ListNode(0);
        dummy->next = head;
        pre1 = dummy;
        pre2 = dummy;
        while(p1 != NULL && p2 != NULL)
        {
            if(p1->val < x){
                p1 = p1->next;
                pre1 = pre1 ->next;
                p2 = p2->next;
                pre2 = pre2->next;
            }
            else{
                while(p2 != NULL)
                {
                    if(p2 -> val < x)
                    {
                        //found the one to insert,insert it and move on
                        pre1->next = p2;
                        pre2->next = p2->next;
                        p2->next = p1;
                        p2 = pre2->next;
                        pre1 = pre1->next;
                        break;
                    }
                    else
                    {
                        p2 = p2->next;
                        pre2 = pre2->next;
                    }
                }
            }
        }
        
        return dummy->next;
    }






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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值