LeetCode 25. K 个一组翻转链表（递归+模拟）

1.题目

给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。
k 是一个正整数，它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

示例：

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

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

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

 

说明：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

递归

递归实质上就是将原问题分解，我们先判断链表元素数量是否达到k，若未达到，直接返回head即可；
当数量大于等于k时，我们先将pre指针移动到要进行翻转的最后一个元素的下一个位置，然后我们就开始翻转第一段链表，写在第二个while中，剩下的部分自然而然就交给reverseKGroup函数去处理，将其返回值接到第一段链表的后面。

class Solution
{
public:
    ListNode* reverseKGroup(ListNode* head, int k)
    {
        ListNode *pre = head;
        int cnt = 0;
        while(pre != nullptr && cnt < k)
        {
            pre = pre -> next;
            cnt++;
        }
        if(cnt < k)
            return head;
        ListNode *cur = nullptr;
        ListNode *p = head;
        while(p != pre)
        {
            ListNode *temp = p -> next;
            p -> next = cur;
            cur = p;
            p =temp;
        }
        head -> next = reverseKGroup(pre,  k);
        return cur;
    }
};

官方模拟

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
            # 查看剩余部分长度是否大于等于 k
            for i in range(k):
                tail = tail.next
                if not tail:
                    return hair.next
            nex = tail.next
            head, tail = self.reverse(head, tail)
            # 把子链表重新接回原链表
            pre.next = head
            tail.next = nex
            pre = tail
            head = tail.next
        
        return hair.next