链表之K个一组翻转链表

在这里插入图片描述
利用递归性质,可以得到大致的算法流程:

1、先反转以 head 开头的 k 个元素。

在这里插入图片描述

2、将第 k + 1 个元素作为 head 递归调用 reverseKGroup 函数

在这里插入图片描述

3、将上述两个过程的结果连接起来。

在这里插入图片描述
整体思路就是这样了,最后一点值得注意的是,递归函数都有个 base case,对于这个问题是什么呢?

题目说了,如果最后的元素不足 k 个,就保持不变。这就是 base case
在这里插入图片描述
在这里插入图片描述

class Solution:
    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
        if not head:
            return head
        a = b = head
        for i in range(k):
            if not b:
                return head
            b = b.next
        newHead = self.reverse(a, b)
        a.next = self.reverseKGroup(b, k)
        return newHead

    
    def reverse(self, a, b):
        pre = None
        curr = a
        nex = a
        while curr != b:
            nex = curr.next
            curr.next = pre
            pre = curr
            curr = nex
        return pre
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值