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

链表分割

时间:2017-04-02 00:07:20      阅读:171      评论:0      收藏:0      [点我收藏+]

标签:排列   class   desc   ext   work   while   nod   tno   strong   

题目描述

编写代码,以给定值x为基准将链表分割成两部分,所有小于x的结点排在大于或等于x的结点之前

给定一个链表的头指针 ListNode* pHead,请返回重新排列后的链表的头指针。注意:分割以后保持原来的数据顺序不变。

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};*/
class Partition {
public:
    ListNode* partition(ListNode* pHead, int x) {
        // write code here
        if(!pHead || !pHead->next)
            return pHead;
        
       
        ListNode* littleBegin = new ListNode(-1);
        ListNode* bigBegin = new ListNode(-1);
        ListNode* little = littleBegin;
        ListNode* big = bigBegin;
        
        ListNode* pWork = pHead;
        while(pWork){
            if(pWork->val < x){
                little->next = pWork;
                little = little->next;
            }else{
                big->next = pWork;
                big = big->next;
            }
            pWork = pWork->next;
        }
        
           little->next = bigBegin->next;
        big->next = nullptr;
        return littleBegin->next;
        
    }
};
//思路为建立两个链表little中存放比x小的元素,big中存放比x大的元素。
//之后合并little和big两个链表。

 

链表分割

标签:排列   class   desc   ext   work   while   nod   tno   strong   

原文地址:http://www.cnblogs.com/xiuxiu55/p/6657804.html

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