面试题2.04. 分割链表

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        //一个链表存储＞x的，一个链表存储≤x的，最后连起来即可
        auto dummy=new ListNode(-1),dummy2=new ListNode(-1);
        auto tail=dummy,tail2=dummy2;
        for(auto p=head;p;){
            auto t=p->next;
            if(p->val<x)    tail=tail->next=p;
            else tail2=tail2->next=p;
            p->next=nullptr;
            p=t;
        }
        tail->next=dummy2->next;
        return dummy->next;
    }
};