leetcode86. 分隔链表

题目：

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

思路：
  把小于x的节点从原链表删除，然后依次串起来，就得到2个链表，一个是小于x的链表，另一个是大于等于x的链表，然后把这俩链表串起来即可。

代码：

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def partition(self, head, x):
        """
        :type head: ListNode
        :type x: int
        :rtype: ListNode
        """
        aim = ListNode(-1)
        aim.next = head
        pre = aim
        first = None
        last = None

        while head:
            if head.val < x:
                temp = head
                head = head.next
                pre.next = head
                temp.next = None

                if not first:
                    first = temp
                    last = temp
                else:
                    last.next = temp
                    last = last.next

            else:
                head = head.next
                pre = pre.next
        if last:
            last.next = aim.next
        return first if first else aim.next