题目链接:链表中的节点每k个一组翻转
解题思路1:递归
代码如下:
ListNode* reverseKGroup(ListNode* head, int k) {
//找到每次反转的尾部,尾部即是反转后的头部
ListNode* tail = head;
//遍历k次找到尾部,一组有k个,从当前节点往后走k部,就走到了第k个,也就是该组的尾部
//本for循环的目的就是找到每一组的头部,进行拼接
for(int i=0; i<k; i++){
if(tail == nullptr){
//如果不足k步走到了链表尾部,直接返回head,不反转,该组顺序不变
return head;
}
tail = tail->next;
}
//链表进行反转的时候需要前序节点和当前节点
//当前节点的next要指向前序节点,当前节点要指向下一个节点
ListNode* pre = nullptr;
ListNode* cur = head;
//在到达当前段尾节点前
while(cur != tail){
ListNode* temp = cur->next;
cur->next = pre;
pre = cur;
cur = temp;
}
//head是原来的头,现在是反转后的尾,用来和下一组进行连接
//为什么返回pre呢?pre是每一个当前组的头节点
head->next = reverseKGroup(tail, k);
return pre;
}
解题思路2:非递归
代码如下:
ListNode* reverseKGroup(ListNode* head, int k) {
//创建一个虚拟节点
ListNode* dummy = new ListNode(0);
//虚拟节点指向头结点
dummy->next = head;
ListNode* pre = dummy;//pre是每组的前一个节点,要和下一组进行连接
ListNode* end = dummy;
while(end->next != nullptr){
//每k个一组进行反转,end是每k个链表的最后一个
for(int i=0; i<k&&end!=nullptr; i++){
end = end->next;
}
//如果end是空的,则说明不够k个,凑不成一组,直接退出循环即可
if(end == nullptr) break;
//反转开始的节点是pre的下一个节点
ListNode* start = pre->next;
//next是下一次反转的头结点
ListNode* next = end->next;
//end是本组的最后一个节点
end->next = nullptr;
//将pre和反转后的k个一组的链表连接起来
pre->next = reverse(start);
start->next = next;//start是下一次反转的头,更新为next
pre = start;
end = start;
}
return dummy->next;
}
//链表的反转
ListNode* reverse(ListNode* head){
ListNode* pre = nullptr;
ListNode* cur = head;
while (cur != nullptr) {
ListNode* temp = cur->next;
cur->next = pre;
pre = cur;
cur = temp;
}
return pre;
}