Leetcode25. K 个一组翻转链表——python求解

最新推荐文章于 2024-12-05 10:30:34 发布

原创最新推荐文章于 2024-12-05 10:30:34 发布 · 256 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#链表 #算法 #leetcode #数据结构 #python

算法编程专栏收录该内容

62 篇文章

订阅专栏

本文详细解析了LeetCode第25题“K个一组翻转链表”的解决方案，通过Python实现，展示了如何在常数额外空间下，将链表每k个节点进行翻转，同时保持最后剩余节点的原有顺序。

Leetcode25. K 个一组翻转链表

编程语言

python

题目

给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。

k 是一个正整数，它的值小于或等于链表的长度。

如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

示例

给你这个链表：1->2->3->4->5

当 k = 2 时，应当返回: 2->1->4->3->5

当 k = 3 时，应当返回: 3->2->1->4->5

说明

你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

来源

链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group

思路

一个函数进行子链的翻转，主函数中进行子链的判断及整合

代码

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

class Solution:
    def reverse(self, head: ListNode, tail: ListNode):
        prev = tail.next
        p = head
        while prev != tail:
            nex = p.next
            p.next = prev
            prev = p
            p = nex
        return tail, head

    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
        hair = ListNode(0)
        hair.next = head
        pre = hair

        while head:
            tail = pre
            for i in range(k):
                tail = tail.next
                if not tail:
                    return hair.next
            head, tail = self.reverse(head, tail)
            pre.next = head
            pre = tail
            head = tail.next
        
        return hair.next