思路:
先找出这k个结点组成的链表,然后利用常规的单链表翻转方法进行翻转。需要一个指针lastEnd记录已经翻转后形成的链表的尾结点,还需要一个指针nextHead记录下一个待翻转的链表的链头,然后需要start和end记录当前需要翻转的k个结点组成的链表的表头和表尾,需要注意的是在调用reverse()进行子链表翻转之前需要把end的next赋值为NULL,然后翻转后新的链表的表头为end,表尾是next。
代码:
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int k) {
ListNode* vhead=new ListNode(-1);
vhead->next=head;
ListNode* lastEnd=vhead;
ListNode* start=NULL;
ListNode* end=NULL;
while(lastEnd->next){
start=lastEnd->next;
end=lastEnd;
for(int i=0;i<k;i++){
end=end->next;
if(!end){
break;
}
}
if(!end){
break;
}
ListNode* nextHead=end->next; //保存下一个待翻转的链表的链头
end->next=NULL; //这里注意要把end的next置为NULL,这样才能在reverse的时候只翻转这个k个结点
lastEnd->next=reverse(start); //上一个翻转后的链尾指向当前翻转后的链表后的链头
lastEnd=start; //翻转后的头结点变成了尾结点,更新已翻转后的链表的尾指针,使其进入下一个while的判断
lastEnd->next=nextHead; //此时start作为尾结点后next为NULL,将它链接到下一个链表的头
}
return vhead->next;
}
ListNode* reverse(ListNode* head){ //返回子链翻转后的链头
ListNode* pre=NULL;
ListNode* cur=head;
ListNode* next=NULL;
while(cur){
next=cur->next;
cur->next=pre;
pre=cur;
cur=next;
}
return pre;
}
};