题目描述
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。 k 是一个正整数,它的值小于或等于链表的长度。 如果节点总数不是 k
的整数倍,那么请将最后剩余的节点保持原有顺序。
题目解析
题目要求在一个链表中以k个链表结点为单位进行反转,什么意思呢?
你可以想象把一个很长的链表分成很多个小链表,每一份的长度都是k
最后一份的长度如果小于k则不需要反转,
然后对每个小链表进行反转,最后将所有反转后的小链表按照之前的顺序拼接在一起
解题思路
参考这位大佬的博客:大佬解题思路
首先,此题是分段翻转链表,那么首先一个翻转链表函数是需要提前写好的,然后我们可以用两个指针将待翻转的链表段的首节点和尾节点确定下来(用于处理与下一段链表的拼接),每次我们取的待翻转链表时,就将这个链表去下一段拆开,即一段一段翻转,翻转完毕后,我们需要做两步拼接,即将翻转的链表与前一段链表拼接,与下一段链表可以先断开着不用拼接,因为我们已经记录了下一段链表的头,待下一段链表翻转完毕后,我们再进行类似前面的操作与前一段链表拼接(当然,如果这段链表不能翻转了,那我们直接将刚才断开的链重新拼接即可)。
下图是我们解此题的模型假设我们即将翻转的链表段是head1-end1,首先我们通过计数找到end1的位置,然后将end1与head2断开,然后翻转head1-end1,然后处理last的next指针(i.e.,处理当前翻转链表段与上一链表段的关系),当处理完毕后,重新初始化计数和推移cur指针进行下一轮的迭代。
完整代码:
//题目解析:
//题目要求在一个链表中以k个链表结点为单位进行反转,什么意思呢?
//你可以想象把一个很长的链表分成很多个小链表,每一份的长度都是k
//最后一份的长度如果小于k则不需要反转,
//然后对每个小链表进行反转,最后将所有反转后的小链表按照之前的顺序拼接在一起
#include<iostream>
#include<vector>
using namespace std;
struct ListNode{
int val;
ListNode *next;
};
ListNode *createByTail(){
ListNode *head;
ListNode *p;
ListNode *tail;
head=NULL;
int len;
cin>>len;
int n=0;
int num;
while(n<len && cin>>num){
p=new ListNode;
p->val=num;
p->next=NULL;
n=n+1;
if(n==1){
head=p;
}else{
tail->next=p;
}
tail=p;
}
return head;
}
void displayLink(ListNode *head){
ListNode *p;
p=head;
cout<<"head-->";
while(p!=NULL){
cout<<p->val<<"-->";
p=p->next;
}
cout<<"tail\n";
}
class Solution{
public:
void reverseList(ListNode *head){
ListNode *prev=NULL;
ListNode *next;
while(head){
next=head->next;
head->next=prev;
prev=head;
head=next;
}
}
ListNode *reverseKGroup(ListNode *head,int k){
ListNode *head1=head;
ListNode *end1;
ListNode *head2;
ListNode *cur=head;
ListNode *last=NULL;
ListNode *res=NULL;
int cnt=1;
while(cur){
head1=cur;
while(cur && cnt%k !=0){
cur=cur->next;
++cnt;
}
if(!cur){
last->next=head1;
break;
}
end1=cur;
head2=end1->next;
end1->next=NULL;
reverseList(head1);
if(head1==head) res=end1;
if(last){
last->next=end1;
last=head1;
}else{
last=head1;
}
cur=head2;
cnt=1;
}
return res;
}
};
int main(){
ListNode *head=createByTail();
int k;
cin>>k;
head=Solution().reverseKGroup(head,k);
displayLink(head);
return 0;
}