【LeetCode-25】K 个一组翻转链表

问题

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

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

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

示例

解答

class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        ListNode* nxt_head = head;
        for (int i = 0; i < k; i++) { // 递归终止条件，顺便找到下一层递归的起始head
            if (!nxt_head) return head;
            nxt_head = nxt_head->next;
        }
        ListNode* newHead = reverse(head, nxt_head);
        head->next = reverseKGroup(nxt_head, k);
        return newHead;
    }
private:
    ListNode* reverse(ListNode* head, ListNode* nxt_head) {
        if (head->next == nxt_head) return head; // 不用担心head为空节点
        ListNode* res = reverse(head->next, nxt_head);
        head->next->next = head;
        return res;
    }
};

重点思路

找好递归最小单元、输入返回值，弄清楚递归的终止条件。