码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 86. 分隔链表

时间:2020-07-12 12:24:42      阅读:45      评论:0      收藏:0      [点我收藏+]

标签:head   lan   __init__   solution   code   大于   tco   while   art   

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

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

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

class Solution:
    def partition(self, head: ListNode, x: int) -> ListNode:
        if head is None or head.next is None:
            return head 
        left_head = ListNode(None)
        right_head = ListNode(None)
        cur_left = left_head
        cur_right = right_head
        cur = head 
        while cur:
            if cur.val < x:
                cur_left.next = cur 
                cur_left = cur_left.next
            else:
                cur_right.next = cur
                cur_right = cur_right.next
            cur = cur.next 
        cur_left.next = right_head.next 
        cur_right.next = None 
        return left_head.next

LeetCode 86. 分隔链表

标签:head   lan   __init__   solution   code   大于   tco   while   art   

原文地址:https://www.cnblogs.com/sandy-t/p/13287425.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!