25. K 个一组翻转链表（Python）

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

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

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

示例：

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

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

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

class Solution:
    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
        # 设置虚拟节点
        dummy = ListNode(0)
        prev = dummy
        start = head
        flag = True
        while head:
            for i in range(k):
                if not head:
                    # 剩余节点数量小于k，跳出
                    flag = False
                    break
                head = head.next
            if not flag:
                break
            # 上次翻转后的节点连接这次翻转后的节点
            prev.next = self.reverse(start,head)
            # 连接这次翻转以后的正常节点
            start.next = head
            # 更新位置
            prev = start
            # 更新位置
            start = head
        return dummy.next

    def reverse(self, start,end) -> ListNode:
        cur, prev = start, None
        while cur!=end:
            cur.next, prev, cur = prev, cur, cur.next
        return prev