题解 | #链表中的节点每k个一组翻转#
链表中的节点每k个一组翻转
http://www.nowcoder.com/practice/b49c3dc907814e9bbfa8437c251b028e
``/**
- struct ListNode {
- int val;
- struct ListNode *next;
- }; */
class Solution { public: /** * * @param head ListNode类 * @param k int整型 * @return ListNode类 */
ListNode* reverseKGroup(ListNode* head, int k) {
// write code here
int m=k;//后序需要递归,k不可变,所以用一个变量替代k
if(k==1||head==nullptr)//k为1时或者head为空时,相当于不反转,直接返回head
{
return head;
}
int count=0;//统计链表节点的数量
ListNode* l=head;
while(l!=nullptr)
{
l=l->next;
count++;
}
if(count<k)//节点数量比k小,则不反转直接返回head
return head;
/*
下面的代码对head的前k个节点进行反转
*/
ListNode* begin=head;//表示反转时的当前节点
ListNode* store=begin;//保存住begin,即保存住head。因为这两个节点后序会变化,不保存,反转后无法完成拼接
ListNode* temp=nullptr;//用于反转时记录当前节点的前一个节点
ListNode* end=head;//用于找到第k个节点
while(--m)
{
end=end->next;
}
ListNode* la=end->next;//la指向第k个节点的下一个节点,用于终止反转的while循环
if(la!=nullptr){
while(head!=la)
{
head=head->next;
begin->next=temp;
temp=begin;
begin=head;
}
}
else{
while(head!=nullptr)
{
head=head->next;
begin->next=temp;
temp=begin;
begin=head;
}
}
/*
前k个节点反转完成后,temp指向了第k个节点,拼接时,它在首端,store保存的为反转前的head,拼接时它在后端,且它的next
指向递归调用的返回值
*/
l=temp;
store->next=reverseKGroup(head,k);
return l;
}
};