[力扣 Hot100]Day31 K 个一组翻转链表

题目描述

给你链表的头节点 head ，每 k 个节点一组进行翻转，请你返回修改后的链表。
k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。
你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。
出处

思路

跟昨天两两交换的思路一致，引入了指针数组来用空间换时间，改进了一下使得主函数更加简洁。

代码

class Solution {
public:
    ListNode* my_swap(ListNode *h, int k){//交换h后的k个结点并返回下一结点的指针
        ListNode** v=(ListNode**)malloc(sizeof(ListNode*)*k);
        ListNode* p=h->next;
        for(int i=0; i<k; i++){
            if(!p) return nullptr;
            v[i]=p;
            p=p->next;
        }
        v[0]->next=v[k-1]->next;
        h->next=v[k-1];
        for(int i=1; i<k; i++){
            v[i]->next=v[i-1];
        }
        return v[0];
    }
    ListNode* reverseKGroup(ListNode* head, int k) {
        ListNode* p=new ListNode(0,head);
        ListNode* q=p;
        while(p) p=my_swap(p,k);
        return q->next;
    }
};