LeetCode-25

25. K 个一组翻转链表

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

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

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

Example

input
1->2->3->4->5，k=2
output
2->1->4->3->5

input
1->2->3->4->5，k=3
output
3->2->1->4->5

Note

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

思路

第一种： 用栈来处理，k个节点入栈，然后出栈形成新的链表，依次k个节点入栈…
第二章： 参考之前的 “交换两个相邻节点” 的题目，递归处理即可

代码如下：

# 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 k == 1:
            return head

        #k个一起处理
        cur = head
        for i in range(k-1):
            if cur is None or cur.next is None:
                return head
            cur = cur.next
        
        cur.next = self.reverseKGroup(cur.next, k)
        #反转k个节点
        res = head.next
        head.next = cur.next
        for i in range(k-1):
            ans = res
            res = res.next
            ans.next = head
            head = ans
        
        return head