LeetCode.25K个一组翻转链表详解

问题描述

给你链表的头节点 head ,每 k个节点一组进行翻转,请你返回修改后的链表。

k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k的整数倍,那么请将最后剩余的节点保持原有顺序。

你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。

问题理解

给定一个链表和一个数字 k,任务是将链表中的每 k 个节点做一次翻转,如果链表的长度不是 k 的整数倍,则剩余的节点保持原有顺序。这是一个典型的链表操作问题,涉及到链表的遍历、节点的重新链接等操作。

解题思路1

1. 确定操作单位

在理解了问题之后,首先明确操作的基本单位是"节点组",每组包含 k 个节点。操作的核心是对这些节点组进行翻转。如果当前组节点不足 k 个,直接跳过该组,保持原样。

2. 理解翻转操作

链表的翻转是基础操作,但在此问题中,我们不是翻转整个链表,而是翻转部分连续的节点。这要求我们能够准确识别每个待翻转的节点组的起始点和终点。

3. 操作过程的细化

为了实现这一操作,我们需要细化操作过程:

  • 定位节点组:从链表头部开始,向后数 k 个节点,标记为一个待翻转的节点组。
  • 执行翻转:对每个节点组执行翻转操作。翻转涉及断开节点的原有连接,然后按逆序重新连接。
  • 处理边界:每次翻转完成后,需要正确处理该节点组与前后节点的连接关系,确保链表的完整性。
4. 算法的选择

为了方便处理链表的头部可能变化的情况,采用了一个虚拟头节点(dummy node)。这个节点在实际链表的头部之前,使得链表头部的处理逻辑与其他部分一致。

5. 连接与继续

完成一个节点组的翻转后,需要将翻转后的尾部(翻转前的头部)连接到下一个节点组的头部。同时,更新操作指针,继续识别和处理下一个节点组。

代码实现

cpp 复制代码
/**
 * 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) {
        if (head == nullptr || head->next == nullptr || k == 1) {
            return head;
        }

        ListNode dummy(0);
        dummy.next = head;
        ListNode* prevGroupEnd = &dummy;

        while (true) {
            // 检查是否有足够的节点来进行翻转
            ListNode* current = prevGroupEnd;
            for (int i = 0; i < k && current != nullptr; i++) {
                current = current->next;
            }
            if (current == nullptr)
                break; // 如果不够 k 个,就不进行翻转

            ListNode* groupStart = prevGroupEnd->next;
            ListNode* nextGroupStart = current->next;

            // 翻转当前的 k 个节点
            ListNode* prev = nextGroupStart;
            ListNode* curr = groupStart;
            while (curr != nextGroupStart) {
                ListNode* temp = curr->next;
                curr->next = prev;
                prev = curr;
                curr = temp;
            }

            // 将翻转后的子链表接回主链表
            prevGroupEnd->next = prev;
            prevGroupEnd = groupStart;
        }

        return dummy.next;
    }
};

进阶

你可以设计一个只用 O(1) 额外内存空间的算法解决此问题吗?

解题思路

1. 链表遍历和长度计算

首先,我们需要遍历整个链表来确定其长度,这样可以判断出能够完整进行多少次k节点的翻转。这一步是必要的,因为如果链表长度未达到k,则不应进行翻转。

2. 翻转链表的k个节点

对于链表翻转操作,我们通常会使用三个指针:prevcurr(当前节点)、nex(当前节点的下一个节点)。通过这三个指针可以实现链表的局部翻转:

  • prev 指针始终指向当前k个节点组的前一个节点,以便在翻转后将翻转的部分重新连接到主链表上。
  • curr 是当前正在处理的节点,也是每次k组翻转的第一个节点。
  • nex 用于存储curr的下一个节点,确保在翻转过程中不会丢失链表的其余部分。

具体翻转步骤是:对于每个k节点组,我们将curr的下一个节点移到prev之后,这样原来的第二个节点就成了新的第一个节点,重复这个操作k-1次,便完成了一组的翻转。

3. 连接翻转后的部分

每次翻转完成后,需要更新prev指针到当前k组的末尾,这样在下一次循环时,它可以正确地指向下一组待翻转节点组的前一个节点。同时,更新剩余节点计数,减去已翻转的k个节点。

4. 处理不足k个的节点

在遍历链表时,如果发现剩余节点数小于k,则不进行翻转,直接将这些节点连接到已翻转部分的末尾。

5. 虚拟头节点的使用

在处理链表问题时,使用虚拟头节点可以大大简化边界条件的处理,例如当链表头部节点需要被移动或删除时。在这个问题中,虚拟头节点帮助我们更方便地处理头k个节点的翻转,以及最终返回新的头节点。

通过这样的步骤,我们可以实现题目要求的每k个节点一组的翻转,而不足k个的部分保持不变。这种方法的时间复杂度为O(n),空间复杂度为O(1),因为我们没有使用额外的数据结构,仅通过修改节点间的链接来达到翻转的目的。

代码实现

cpp 复制代码
/**
 * 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) {
        ListNode* dummy = new ListNode(0);
        dummy->next = head;
        ListNode *prev = dummy, *curr = head, *nex = nullptr;

        // 首先计算链表长度
        int count = 0;
        while (curr) {
            curr = curr->next;
            count++;
        }

        // 当链表中至少有k个节点时进行翻转
        while (count >= k) {
            curr = prev->next;
            nex = curr->next;
            for (int i = 1; i < k; i++) {
                curr->next = nex->next;
                nex->next = prev->next;
                prev->next = nex;
                nex = curr->next;
            }
            prev = curr;
            count -= k;
        }

        ListNode* ret = dummy->next;
        delete dummy; // 释放虚拟头节点
        return ret;
    }
};
相关推荐
何曾参静谧几秒前
「C/C++」C/C++ 之 变量作用域详解
c语言·开发语言·c++
AI街潜水的八角10 分钟前
基于C++的决策树C4.5机器学习算法(不调包)
c++·算法·决策树·机器学习
JSU_曾是此间年少39 分钟前
数据结构——线性表与链表
数据结构·c++·算法
sjsjs111 小时前
【数据结构-合法括号字符串】【hard】【拼多多面试题】力扣32. 最长有效括号
数据结构·leetcode
此生只爱蛋2 小时前
【手撕排序2】快速排序
c语言·c++·算法·排序算法
blammmp2 小时前
Java:数据结构-枚举
java·开发语言·数据结构
何曾参静谧2 小时前
「C/C++」C/C++ 指针篇 之 指针运算
c语言·开发语言·c++
昂子的博客2 小时前
基础数据结构——队列(链表实现)
数据结构
咕咕吖2 小时前
对称二叉树(力扣101)
算法·leetcode·职场和发展
九圣残炎3 小时前
【从零开始的LeetCode-算法】1456. 定长子串中元音的最大数目
java·算法·leetcode