5.5 链表分化

题目:

对于一个链表,我们需要用一个特定阈值完成对它的分化,使得小于等于这个值的结点移到前面,大于该值的结点在后面,同时保证两类结点内部的位置关系不变。

给定一个链表的头结点head,同时给定阈值val,请返回一个链表,使小于等于它的结点在前,大于等于它的在后,保证结点值不重复。

测试用例:

input:{1,4,2,5},3
output:{1,2,4,5}

思路:

 用两个链表分别保存小于等于val的结点和大于val的结点,最后拼接起来即可。注意链表末尾是需要取空的,否则会成环。

代码:

# -*- coding:utf-8 -*-
class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

class Divide:
    def listDivide(self, head, val):
        # write code here
        equal = ListNode(0) #记录小于等于val的结点的链表
        e_cur = equal #equal的尾结点
        great = ListNode(0) #记录大于val的结点的链表
        g_cur = great #great的尾结点
        while head:
            if head.val <= val:
                e_cur.next = head
                e_cur = e_cur.next
            else:
                g_cur.next = head
                g_cur = g_cur.next
            head = head.next
        e_cur.next = great.next
        g_cur.next = None #最后一个结点的next需要置空
        return equal.next

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值