Leetcode K个一组翻转链表

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;
    }
};

定义一个双向队列用于存放中间交换节点组，当还有剩余节点时，就像栈一样出栈，然后连接每一个节点，当剩余节点不足以整除时，就要像队列一样出队，然后连接每一个节点，最终完善细节。