[LintCode 96] 链表划分(Python)

题目描述

给定一个单链表和数值x，划分链表使得所有小于x的节点排在大于等于x的节点之前。
你应该保留两部分内链表节点原有的相对顺序。

样例
给定链表 1->4->3->2->5->2->null，并且 x=3
返回 1->2->2->4->3->5->null

思路

新建两个链表，一个存储小于x的节点，一个存储大于等于x的节点。最后把两个链表连接起来。注意把最后一个节点的next置为None，否则可能出现死循环。

代码

"""
Definition of ListNode
class ListNode(object):
    def __init__(self, val, next=None):
        self.val = val
        self.next = next
"""


class Solution:
    """
    @param: head: The first node of linked list
    @param: x: An integer
    @return: A ListNode
    """