链表中的节点每k个一组翻转

• 描述

将给出的链表中的节点每 k 个一组翻转，返回翻转后的链表
如果链表中的节点数不是 k 的倍数，将最后剩下的节点保持原样
你不能更改节点中的值，只能更改节点本身。

 

数据范围：  0≤n≤2000 ， 1≤k≤2000 ，链表中每个元素都满足 0≤val≤1000
要求空间复杂度 O(1)，时间复杂度 O(n)

 

例如：

给定的链表是 1→2→3→4→5

对于 k=2 , 你应该返回 2→1→4→3→5

对于 k=3 , 你应该返回 3→2→1→4→5

 

• 算法思想

此题使用递归较为方便

step 1：每次从进入函数的头节点优先遍历链表k次，分出一组，若是后续不足k个节点，不用反转直接返回头。

step 2：从进入函数的头节点开始，依次反转接下来的一组链表，反转过程同反转链表的题目。

step 3：这一组经过反转后，原来的头变成了尾，后面接下一组的反转结果，下一组采用上述递归继续。

具体过程如下图（k=3为例）：

 

递归过程

 

 

 

 

 

 

 

 

 

回溯过程

 

 

 

 

 

 

最底层返回的是第一个分组的pre,也就是新链表的头结点。

 

• 代码

 

/**
 * struct ListNode {
 *    int val;
 *    struct ListNode *next;
 * };
 */

#include <cstddef>
class Solution {
public:
    /**
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* reverseKGroup(ListNode* head, int k) {
        ListNode *tail=head;
        //遍历k次
        for(int i=0;i<k;i++){
            if(tail==nullptr) return head;
            tail=tail->next;
        }
        //反转当前段
        ListNode *pre=nullptr;
        ListNode* cur=head;
        while(cur!=tail){
            ListNode* temp=cur->next;
            cur->next=pre;
            pre=cur;
            cur=temp;
        }
        //递归
        head->next=reverseKGroup(tail,k);
        //返回当前段的pre,用来连接前一段的链表
        return pre;
    }
};

 

 参考：https://blog.nowcoder.net/n/7a045d18a2b049379ca97f6b408f2691?f=comment