题目描述
给你链表的头节点 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;}
};