k个一组反转链表
解题思路
暴力求解:k个一组进行翻转,每一次翻转之前检查剩下的结点是否有k个,如果不够k个,就不进行翻转。
用指针pre 记录每次翻转开始前的第一个结点,因为每次翻转前的第一个结点也就是下一次翻转要接入的位置。 翻转链表采取头插法。
代码
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int n) {
ListNode* h = new ListNode;
ListNode* p = head, *q = head;
ListNode* k = h, *pre = nullptr;
while(q){
ListNode* tmp = q;
for(int i = 0;i < n - 1;++i){
tmp = tmp->next;
if(!tmp){
k->next = q;
return h->next;
}
}
for(int i = 0;i < n;++i){
if(i == 0)pre = q;
p = q->next;
q->next = k->next;
k->next = q;
q = p;
}
k = pre;
}
return h->next;
}
};
|