题目
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
示例:
给你这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
代码:
时间O(n),空间O(1)
有点复杂的模拟,用虚拟头指针统一
/**
* 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* reversek(ListNode* head,int k){
ListNode *p1=head->next,*p2=head->next->next;
for(int i=1;i<k;i++){
ListNode *tmp=p2->next;
p2->next=p1;
p1=p2;
p2=tmp;
}
head->next->next=p2;
return p1;
}
ListNode* reverseKGroup(ListNode* head, int k) {
ListNode *vhead=new ListNode(0,head);
ListNode *cur=vhead;
bool havek=true;
while(1){
ListNode *tmp=cur;
for(int i=0;i<k;i++){
if(!tmp->next){
havek=false;
break;
}
else tmp=tmp->next;
}
if(havek){
tmp=cur->next;
cur->next=reversek(cur,k);
cur=tmp;
}
else break;
}
cur=vhead->next;
delete vhead;
return cur;
}
};