leetcode Partition List

最新推荐文章于 2019-07-16 09:33:43 发布

Do_Cool_Thing

最新推荐文章于 2019-07-16 09:33:43 发布

阅读量331

点赞数

分类专栏： leetcode Fighting

本文链接：https://blog.csdn.net/menyangyang/article/details/38828127

版权

leetcode Fighting 专栏收录该内容

133 篇文章 0 订阅

订阅专栏

开辟两个链表分别存放小于和大于等于x的结点

代码

class Solution {
public:
    ListNode *partition(ListNode *head, int x) {
       
        if(head==NULL)
            return head;
            
        ListNode * smallHead = new ListNode(0);
        ListNode * bigHead = new ListNode(0);
        ListNode * p = head;
        ListNode *q1, *q2;
        q1 = smallHead;
        q2 = bigHead;
        
        while(p!=NULL)
        {
            ListNode * tempNode = new ListNode(p->val);
            if(p->val<x)
            {
                q1->next = tempNode;
                q1 = tempNode;
            }
            else
            {
                q2->next = tempNode;
                q2 = tempNode;
            }
            
            p = p->next;
        }
        
        q1 = smallHead; 
        while(q1->next!=NULL)
            q1 = q1->next;
            
        q1->next = bigHead->next;
 
        
        return smallHead->next;
        
    }
};