力扣HOT100(31)K 个一组翻转链表

整体大思路:

把链表按 k 个节点分成一组一组,对每一组单独翻转,然后把翻转后的组和前后的组重新连接起来,最后不足 k 个的组不翻转。

我们只需要解决 3 个小问题:

  1. 怎么判断剩下的节点够不够 k 个?
  2. 怎么翻转一组 k 个节点?
  3. 怎么把翻转后的组和前后的组连接起来?

步骤 1:判断剩余节点是否够 k 个

我们用一个tail指针,从当前组的前驱节点pre开始,往前走 k 步:

  • 如果能走完 k 步,说明剩余节点够 k 个,可以翻转
  • 如果走不到 k 步就到nullptr了,说明不够,直接结束

步骤 2:翻转一组 k 个节点

这是 206 题「反转链表」的变种,我们需要翻转从head到tail的子链表,并且返回翻转后的新头和新尾,方便后面连接。

步骤 3:把翻转后的组重新连接起来

这是这道题最关键的一步,翻转前一定要先保存下一组的头节点,不然会断链!

固定 4 步连接逻辑:

  1. 保存下一组的头:ListNode* nex = tail->next;
  2. 翻转当前组,得到新头和新尾:tie(newHead, newTail) = reverse(head, tail);
  3. 上一组的尾指向新头:pre->next = newHead;
  4. 新尾指向下一组的头:newTail->next = nex;

然后更新指针,准备处理下一组:

  • pre = newTail;(下一组的前驱是当前组的新尾)
  • head = nex;(下一组的头是之前保存的 nex)
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:
    //写一个pair类型的 函数 翻转一个子链表,并返回新的头和尾
    //pair容器用于存放成对出现的 可以返回新生成的pair.first pair.second
    pair<ListNode*,ListNode*> myReverse(ListNode* head,ListNode* tail){
        ListNode* prev = tail->next;
        ListNode* p = head;
        while(prev != tail){
            //反转链表
            ListNode* nex = p->next;//p的下一个存在nex中
            p->next = prev;
            prev = p;
            p = nex;

        }
        return {tail,head};//返回新的头 和尾巴
    }

    ListNode* reverseKGroup(ListNode* head, int k) {
        ListNode* hair = new ListNode (-1,head);
        ListNode* pre =  hair;

        while(head){
            ListNode* tail  = pre;//尾指针 用来判断剩余部分还有k个吗。初始在虚拟头
            for(int i = 0; i<k;i++){
                tail = tail->next;//tail一直往前走
                if(tail == nullptr){//走到空说明不满足k步了 直接返回 终止循环
                    return hair->next;
                }
            }
            //现在就是满足k步的
            ListNode * nex = tail->next;//tail现在的位置在第k个元素
            //接下来调用函数 进行翻转
            
           tie(head, tail) = myReverse(head, tail);


            //子链表重新接回去
            pre->next = head;
            tail->next = nex;
            pre = tail;//用来存放下一组头的前驱
            head = tail->next;//开始前移
        }
        return hair->next;
    }
};
相关推荐
yi01143 分钟前
DAY17: LeetCode 139|单词拆分:从“把单词删掉”到用 DP 记录合法切口
人工智能·笔记·python·算法·leetcode·动态规划
陈希瑞1 小时前
LongCat-2.5-Preview 的web逆向能力实测
人工智能·算法·wasm
tangguofeng2 小时前
基于分治策略的循环赛日程表问题的思路讲解
算法
起个名字费劲死了3 小时前
VisionMaster集成深度学习算法(基础狗)
人工智能·深度学习·算法
lengxuenong3 小时前
枚举例题分析
c++·算法·青少年编程
Navigator_Z3 小时前
LeetCode //C - 1269. Number of Ways to Stay in the Same Place After Some Steps
c语言·算法·leetcode
2601_965742223 小时前
短视频脚本创作方法分享:本地生活类账号的起步思路
大数据·人工智能·算法·ai·新媒体运营·生活
55873生态系统手记3 小时前
读完 55873 资产钱包的技术栈,我重新理解了 "合规钱包" 应该长什么样
算法·架构
旋生万物3 小时前
素数螺旋映射 $z_n=n^{1+i}$ 的角分布统计检验与零模型对比
大数据·前端·人工智能·算法·云原生·螺旋生成论·螺旋相位
HRTOS3 小时前
HRTOS Shell继续完善:优化printf缓冲区与字符串越界保护,国庆后随新版本更新
运维·单片机·嵌入式硬件·算法·51单片机