LeetCode打卡: 86. 分隔链表

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

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

示例:

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

链接:https://leetcode.com/problems/partition-list/
解题思路:
此题类似于数组中将元素以k为界线分放,思路相同,采用双指针。慢指针记录待插位置,快指针寻找待插结点。
我的代码:

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution(object):
    def partition(self, head, x):
        """
        :type head: ListNode
        :type x: int
        :rtype: ListNode
        """
        if head and head.next:
            fast = ListNode(-1)
            fast.next = head
            pre2 = head = fast
            fast = head.next
            while fast and fast.val < x:
                pre2 = fast
                fast = fast.next
            if fast and fast.next:
                pre1 = pre2
                slow = fast
                pre2 = fast
                fast = fast.next
            else:
                return head.next
            while fast:
                if fast.val < x:
                    temp = fast.next
                    pre1.next = fast
                    fast.next = slow
                    pre1 = fast
                    pre2.next = temp
                    fast = temp
                else:
                    pre2 = fast
                    fast = fast.next
            
            head = head.next
            
        return head

Time: O(N)
Space: O(1)

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值