leetcode 86. Partition List【分裂ListNode】

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

Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

Example:

Input: head = 1->4->3->2->5->2, x = 3
Output: 1->2->2->4->3->5

将比x大的放在左边，大于等于x的放在右边

开始想到了是分成两个ListNode

写啊写啊

各种逻辑错误

最后更无语的是 flag变量设置bool类型 还赋值-1 发现怎么也没变啊

 

/**
 * 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==NULL || head->next==NULL)
            return head;
        ListNode *Rt=new ListNode(-1);
        ListNode *R=Rt;
        ListNode *q=new ListNode(-1);
        q->next=head;
        int flag=1;
        if(head->val < x)
            flag=0;
        while(q->next!=NULL){
            if(q->next->val >= x){
                R->next=q->next;
                q->next=q->next->next;
                R=R->next;
                R->next=NULL;
            }
            else{
                q=q->next;
                if(1==flag){
                    flag=-1;
                    head=q;
                }
            }
            //cout<<flag<<endl;
        }
        
        if(flag==1)
            return Rt->next;
        q->next=Rt->next;
        return head;
    }
};