【c++】【leetcode25】k个一组反转链表

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;
		//指针k代表下一次接的位置
        while(q){
            ListNode* tmp = q;//q指向下一次开始翻转的结点
            //先跑k次，看看够不够翻转
            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;//pre指向下一次翻转的结点的前一个结点
                //翻转链表
                p = q->next;
                q->next = k->next;
                k->next = q;
                q = p;
            }
            k = pre;//pre指向下一次翻转的结点的前一个结点
        }
        return h->next;
    }
};