5_5链表的分化

2017-09-14  本文已影响5人  X_Y

对于一个链表,我们需要用一个特定阈值完成对它的分化,使得小于等于这个值的结点移到前面,大于该值的结点在后面,同时保证两类结点内部的位置关系不变。

给定一个链表的头结点head,同时给定阈值val,请返回一个链表,使小于等于它的结点在前,大于等于它的在后,保证结点值不重复。

测试样例:
输入:{1,4,2,5},3
输出:{1,2,4,5}

/*
   struct ListNode {
   int val;
   struct ListNode *next;
   ListNode(int x) : val(x), next(NULL) {}
   };*/
class Divide {
    public:
        ListNode* listDivide(ListNode* head, int val) {
            // write code here
            if(NULL == head){
                return NULL;
            }
            ListNode *left_a = NULL, *right_a = NULL;
            ListNode *left_e = NULL, *right_e = NULL;
            // 如果小于就放到左边的链表,如果大于就放到右边的链表
            while(head != NULL){
                if(head->val <= val){
                    if(NULL == left_a){
                        left_a = head;
                        left_e = head;
                    }else{
                        left_e->next = head;
                        left_e = head;
                    }
                }else{
                    if(NULL == right_a){
                        right_a = head;
                        right_e = head;
                    }else{
                        right_e->next = head;
                        right_e = head;
                    }
                }
                head = head->next;
            }
            // 合并两个链表
            if(NULL != left_e){
                if(NULL != right_a){
                    left_e->next = right_a;
                    right_e->next = NULL;
                }
                //cout<< 1 << endl;
                return left_a;
            }else{
                //cout<< 2 << endl;
                return right_a;
            }
        }
};
上一篇 下一篇

猜你喜欢

热点阅读