K个一组翻转链表
题目描述:
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
说明:
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int k) {
deque<ListNode* > mid;
bool flag = true; // 完整标记
ListNode* current = head; // 当前的节点
ListNode* pre; // k个之前的节点
while(flag){
if(current == head){ // 第一个
// 保存连接点
pre = current;
// 将k个节点推入mid中
for(int i = 0 ; i<k ; i++){
if(current == NULL){
flag = false;
break;
}
mid.push_back(current);
current = current->next;
}
// 将k个节点弹出mid
if(flag){
head = mid.back();
mid.pop_back();
ListNode* temp = head;
while(!mid.empty()){
temp->next = mid.back();
mid.pop_back();
temp = temp->next;
}
}else{ // 剩余的链表长度
head = mid.front();
mid.pop_front();
ListNode* temp = head;
while(!mid.empty()){
temp->next = mid.front();
mid.pop_front();
temp = temp->next;
}
}
}else{
ListNode* temp1 = pre; // 记录前一个连接点
pre = current; // 更新下一个连接点
// 将k个节点推入mid中
for(int i = 0 ; i<k ; i++){
if(current == NULL){
flag = false;
break;
}
mid.push_back(current);
current = current->next;
}
// 将k个节点弹出mid
if(flag){
temp1->next = mid.back();
mid.pop_back();
ListNode* temp2 = temp1->next;
while(!mid.empty()){
temp2->next = mid.back();
mid.pop_back();
temp2 = temp2->next;
}
}else{
if(mid.empty()){
temp1->next = NULL; // 防止闭环
break; // 当中间为空,则说明已经没有元素了
}
temp1->next = mid.front();
mid.pop_front();
ListNode* temp2 = temp1->next;
while(!mid.empty()){
temp2->next = mid.front();
mid.pop_front();
temp2 = temp2->next;
}
}
}
}
return head;
}
};
定义一个双向队列用于存放中间交换节点组,当还有剩余节点时,就像栈一样出栈,然后连接每一个节点,当剩余节点不足以整除时,就要像队列一样出队,然后连接每一个节点,最终完善细节。