86. Partition List 解题记录

题目描述：

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.

For example,
Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.

题目要求是分割一个链表，将小于x的节点放在前面，大于x的节点放在后面，而且这两部分都要按照原来的顺序排列。

解题思路：

既然要求是分割链表成两个部分，那么我们就可以把这两个部分先分开储存，最后再结合在一起。

代码：

/**
 * 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)
            return head;
        ListNode *head1 = new ListNode(0), *head2 = new ListNode(0), *p1 = head1, *p2 = head2; //p1和p2指针用于遍历链表储存两个部分，head1和head2用于存储开头
        while(head){
        //遍历链表
            if(head->val < x){
            //小于的部分
                p1->next = head;
                p1 = head;
            }
            else{
            //大于的部分
                p2->next = head;
                p2 = head;
            }
            head = head->next;
        }
        p1->next = head2->next;//将小于部分的末尾连接大于部分的开头
        p2->next = NULL;//大于部分的结果给上一个NULL
        return head1->next;
    }
};