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

111 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

问题

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

思路

双指针确定反转区间,然后反转链表,返回新的头结点,原来的头结点经过反转后就变成了尾节点。每次递归调用,将 k 个节点反转,并返回反转过的新的头结点。

代码实现

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

class Solution {
public:
    /**
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* reverseKGroup(ListNode* head, int k) {
        // write code here
        ListNode* tail = head;
        // 向前走 k 步,确定反转区间
        for (int i = 0; i  < k; i++)
        {
            if (!tail) return head;
            tail = tail->next;
        }
        // 区间反转,返回新的头节点
        ListNode* newHead = reverse(head, tail);
        // 经过反转 head 变成 尾节点
        head->next = reverseKGroup(tail, k);
        return newHead;
        
    }
    ListNode* reverse(ListNode* head, ListNode* tail)
    {
        ListNode* pre = NULL;
        ListNode* cur = head;
        // 链表反转,返回 tail 前一个节点
        while (cur != tail)
        {
            ListNode* next = cur->next;
            cur->next = pre;
            pre = cur;
            cur = next;
        }
        return pre;
    }
};