Python算法进阶 - 5(分割链表)

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

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

示例:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-list
 

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

class Solution(object):
    def partition(self, head, x):
        before = before_head = ListNode(0)
        after = after_head = ListNode(0)

        while head:
            if head.val < x:
                before.next = head
                before = before.next
            else:
                after.next = head
                after = after.next

            head = head.next
        after.next = None
        before.next = after_head.next
        return before_head.next

这道题,自己虽然想到了大概,但是方法太low超时了,看下官方大神写的,借鉴思路,首先是创建一个哑节点(也及时L.nil)。

灵活运用,以前只局限于双链表才会有哑节点。

好处在于:为了算法实现更容易,我们使用了哑节点初始化。不能让哑节点成为返回链表中的一部分,因此在组合两个链表时需要向前移动一个节点。

思路:

用两个指针创建两个链表,小于x占一个链表,其他的是另一个链表。当第二个链表没有节点的时候,就可以将原来的两个链表进行连接起来就可以了。  

 

 

 

see you  细品!!!

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值