leetcode 25. reverse-nodes-in-k-group K 个一组翻转链表 python3

时间：2021-2-16

题目地址：https://leetcode-cn.com/problems/reverse-nodes-in-k-group/

题目难度：Hard

题目描述：

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

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

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

示例：

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

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

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

说明：

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

---

思路1：迭代

1. 反转整个链表 while cur != None
2. 反转链表 a到b
3. 考虑不需要反转的，递归处理即可

代码段1：通过

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
        if head == None:
            return None
        a = b = head
        for i in range(k):
            if b == None:
                return head
            b = b.next
        
        def reverseN(head, last):
            pre, cur, nxt = None, head, head
            while cur != last:
                nxt = cur.next
                cur.next = pre
                pre = cur
                cur = nxt
            return pre

        newHead = reverseN(a, b)
        a.next = self.reverseKGroup(b, k)
        return newHead

总结：

1. 熟练基础的数据结构
2. 逐步拆解问题