[25]K 个一组翻转链表

//给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。 
//
// k 是一个正整数，它的值小于或等于链表的长度。 
//
// 如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。 
//
// 
//
// 示例： 
//
// 给你这个链表：1->2->3->4->5 
//
// 当 k = 2 时，应当返回: 2->1->4->3->5 
//
// 当 k = 3 时，应当返回: 3->2->1->4->5 
//
// 
//
// 说明： 
//
// 
// 你的算法只能使用常数的额外空间。 
// 你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。 
// 
// Related Topics 链表 
// 👍 915 👎 0

/*
* 25 K 个一组翻转链表
* 2021-02-24 11:13:31
* @author oxygenbytes
*/ 
#include "leetcode.h" 
//leetcode submit region begin(Prohibit modification and deletion)
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        if(!head || !head->next) return head;

        auto ptr = head;
        for(int i = 0;i < k;i++){
            if(!ptr) return head;
            ptr = ptr->next;
        }

        auto cur = head;
        ListNode* pre = nullptr;

        for(int i = 0;i < k;i++){
            auto temp = cur->next;
            cur->next = pre;
            pre = cur;
            cur = temp;
        }

        head->next = reverseKGroup(cur, k);
        return pre;
    }
};
//leetcode submit region end(Prohibit modification and deletion)