题解 | #链表中的节点每k个一组翻转#

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    pair<ListNode*, ListNode*> reverseK(ListNode* head, ListNode* tail){
	  //反转链表，且返回反转后的头尾节点，则需要多使用一个p指针，不能复用head指针
        ListNode* next=nullptr, *pre=nullptr, *p = head;
        while(pre!=tail){
            next = p->next;
            p->next = pre;
            pre = p;
            p = next;
        }
        return {pre,head};
    }
    ListNode* reverseKGroup(ListNode* head, int k) {
        ListNode* dummy = new ListNode(-1);
        dummy->next = head;
        ListNode* pre = dummy, *tail=dummy;
	  // p指向某段的头节点
        ListNode* p = head, *next=nullptr;
        while(p!=nullptr){
            // 找某段的尾节点
            for(int i=0; i<k; ++i){
                if(tail->next) tail = tail->next;
                else   return dummy->next;
            }
		  // 先记录下一个节点
            next = tail->next;
		  // 翻转
            tie(p, tail) = reverseK(p, tail);
		  // 拼接
            pre->next = p;
            tail->next = next;
		  // 下一个
            pre = tail;
            p = next;
        }
        return dummy->next;
    }
};