Partition List(C++分隔鏈表)

解題思路:

(1)兩個鏈表,一個保存較小值,一個保存較大值

(2)最後較小值鏈表的尾指針指向較大值鏈表的頭指針

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode *p=NULL,*q=NULL;
        ListNode *proot=NULL,*qroot=NULL;
        while(head) {
            if(head->val<x) {
                if(p!=NULL) {
                    p->next = head;
                    p = p->next;
                } else {
                    proot = head;
                    p = head;
                }
            } else {
                if(q!=NULL) {
                    q->next = head;
                    q = q->next;
                } else {
                    qroot = head;
                    q = head;
                }
            }
            head = head->next;
        }
        if(proot!=NULL) {
            p->next = qroot;
            if(qroot!=NULL) {
                q->next = NULL;
            }
            return proot;
        } else return qroot;
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章