2023年第一道leetcode🎉
1.题目
给你链表的头节点 head ,每 k 个节点一组进行翻转,请你返回修改后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
2.示例
1)示例 1:
输入:head = [1,2,3,4,5], k = 2
输出:[2,1,4,3,5]
2)示例 2:
输入:head = [1,2,3,4,5], k = 3
输出:[3,2,1,4,5]
3.分析
这道题目是206的加强版,206中没有k的倍数的要求,那么我们要在此基础上考虑k的要求。在上一道题目q,cur,last三个记录当前以及上一个的指针基础上,还要增加st,ed两个指针表示上一段的头和尾来承接每k断之间的连接关系。此外,还要考虑链表的最后可能凑不够k不需要反转的需求。
4.代码
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int k) {
ListNode* cur=head, *last=NULL, *q=NULL, *st=NULL,*ed=NULL,*ans=head;
int op=1,n=0,flag=0,flag2=0;
while(cur!=NULL){
n++;cur=cur->next;
}
if(n<k||k==1) return head;
cur=head;n=n-n%k;
while(cur!=NULL){
if(op==n+1) {st->next=cur;break;}
if(op%k==0){
if(st!=NULL) st->next=cur;st=ed;
if(flag==0) {ans=cur;flag=1;}
}
if(op%k==1) {ed=cur;q=cur->next;cur->next=NULL;last=cur;cur=q;op++;continue;}
q=cur->next;
cur->next=last;
last=cur;
cur=q;
op++;
}
return ans;
if(st!=NULL) st->next=cur;
return ans;
}
};