码迷,mamicode.com
首页 > 其他好文 > 详细

[leetcode]Partition List

时间:2014-12-03 00:27:55      阅读:141      评论:0      收藏:0      [点我收藏+]

标签:leetcode   算法   

问题描述:

Given a linked list and a value x, partition it such that all nodes less thanx come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

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

基本思路:

类似链表快排算法的partition。

代码:

ListNode *partition(ListNode *head, int x) {  //C++
        if(head == NULL|| head->next == NULL)
            return head;
        
        ListNode tmphead(0);
        ListNode* result = &tmphead;
        
        result->next = head;
        ListNode* p = head;
        ListNode* pre = result;
        ListNode* tail = result;
        while(p != NULL)
        {
            if(p->val < x )
            {
                if(p == tail->next)
                {
                    pre = p;
                    p = p->next;
                    tail = tail->next;
                    
                    continue;
                }
                ListNode* tmp = p->next;
                ListNode* tmptail = tail->next;
                tail->next = p;
                p->next = tmptail;
                pre->next = tmp;
                
                tail = tail->next;
                p = tmp;    
                
            }
            else 
            {
                pre = p;
                p = p->next;
                
            }
            
        }
        return result->next;
            
    }


[leetcode]Partition List

标签:leetcode   算法   

原文地址:http://blog.csdn.net/chenlei0630/article/details/41677575

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!