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

整体大思路:

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

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

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

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

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

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

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

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

步骤 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;
    }
};
相关推荐
玖釉-3 小时前
下一个排列:从字典序到原地算法的完整推导
数据结构·c++·windows·算法
IronMurphy3 小时前
【算法五十】62. 不同路径
算法
影寂ldy3 小时前
C#一维数组
算法
过期动态4 小时前
【LeetCode 热题 100】移动零
java·数据结构·算法·leetcode·职场和发展·rabbitmq
计算机安禾4 小时前
【算法分析与设计】第10篇:下界理论与NP完全性初步
大数据·人工智能·算法
水木流年追梦5 小时前
大模型入门-大模型分布式训练2
开发语言·分布式·python·算法·正则表达式·prompt
sali-tec6 小时前
C# 基于OpenCv的视觉工作流-章78-KRT测量
图像处理·人工智能·数码相机·opencv·算法·计算机视觉
菜菜的顾清寒6 小时前
力扣HOT100(32)二叉树的中序遍历
数据结构·算法·leetcode
x2c6 小时前
数据结构:线性表中链表的建立和基本操作(C)
算法