LeetCode86：分隔链表

/*给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
*/
/*
遍历一遍链表，将<x的放在链表1中，将>x的放在链表2中，最后将两个链表合并就好了
*/
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode *list1=new ListNode(0);//记录小于x的节点
        ListNode *p1=list1;
        
        ListNode *list2=new ListNode(0);//记录大于x的节点
        ListNode *p2=list2;
        
        while(head!=NULL)
        {
            if(head->val<x)
            {
                p1->next=head;
                p1=head;
            }
            else
            {
                p2->next=head;
                p2=head;
            }
            head=head->next;
        }
        p2->next=NULL;//循环完成后记得将较大的链表的尾置为null
        p1->next=list2->next;//此时p1指向的是第一个链表的尾，将p1的next指向第二个链表的头就好了，由于list2的第一个节点是0，所以为list2->next
        return list1->next;//最后返回第一个链表的第一个节点
    }
};