Day12 [LeetCode中等] 86. 分隔链表

Day12: [LeetCode中等] 86. 分隔链表

题源：

来自leetcode题库：

https://leetcode-cn.com/problems/partition-list/

思路：

思路就是构造两个链表min和max，分别储存小于x和大于等于x的节点，最后连接两个链表。需要注意的是，当某个链表无节点的时候，要特殊处理一下，总之就是注意边界条件。

代码：

dirty code凑合看吧

/**
 * 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) {
        if(!head||head->next==NULL) return head;
        ListNode* p=NULL,*q=NULL,*p_r=NULL,*q_r=NULL,*s=head;
        while(s){
            if(s->val<x){
                if(p_r==NULL){
                    p=s;
                    p_r=s;
                }else{
                    p_r->next=s;
                    p_r=p_r->next;
                }
            }else{
                if(q_r==NULL){
                    q=s;
                    q_r=s;
                }else{
                    q_r->next=s;
                    q_r=q_r->next;
                }
            }
            s=s->next;
        }
        if(!p_r) return q;
        else p_r->next=q;
        if(!q_r) return p;
        else q_r->next=NULL;
        return p;
    }
};