K 个一组翻转链表
在面试中,链表问题总是一个热门话题,而反转链表又是链表问题中的一个经典问题。今天,我们要解决的是其中最为复杂的变种:反转链表中每k个节点为一组的子链表。
首先是题目链接
描述
给你链表的头节点 head ,每 k 个节点一组进行翻转,请你返回修改后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
示例1:
示例2:
算法思路是通过递归和迭代的方式,将链表分成若干组,每组包含k个节点,然后对每组进行翻转,最后将翻转后的各组重新连接起来,形成一个新的链表。
解决步骤:
1. 初始化翻转尾部指针
我们首先定义一个tail指针,它用于遍历链表直到第k个节点。如果链表中的节点数不足k个,tail将指向链表的末尾,这时我们不需要进行翻转。
ListNode* tail = head;
for (int i = 0; i < k; i++) {
if (tail == NULL)
return head;
tail = tail->next;
}
2. 翻转操作的准备
在确定了翻转的尾部之后,我们定义了两个指针pre和cur,分别指向翻转操作中的前一个节点和当前节点。初始时,pre被设置为NULL,因为翻转的起始节点的前一个节点是不存在的。
ListNode* pre = NULL;
ListNode* cur = head;
3. 逐个节点翻转
接下来是翻转操作的核心。我们使用一个while循环,当cur不等于tail时,即还没有翻转到预定的尾部,我们执行翻转操作。这里,我们通过一个临时指针temp来保存当前节点的下一个节点,然后更新当前节点的next指针指向pre,接着将pre和cur向前移动。
while (cur != tail) {
ListNode* temp = cur->next;
cur->next = pre;
pre = cur;
cur = temp;
}
这个过程实际上是在逐个节点地将链表的一部分翻转,使得cur的下一个节点指向它前面的节点,直到我们到达预定的尾部。
4. 递归处理剩余链表
翻转完当前的k个节点后,我们需要处理剩余的链表。我们通过递归调用reverseKGroup函数,将翻转后的尾部节点tail作为新的头节点,继续执行翻转操作。
head->next = reverseKGroup(tail, k);
这里,head是翻转后的子链表的头节点,它的next指针现在指向了下一段需要翻转的链表。
5. 返回翻转后的头节点
最后,我们返回翻转后的子链表的头节点pre,它指向了翻转后的链表的第一个节点。
return pre;
完整代码如下:
class Solution {
public:
ListNode* reverseKGroup(ListNode* head, int k) {
//找到每次翻转的尾部
ListNode* tail = head;
//遍历k次到尾部
for (int i = 0; i < k; i++) {
//如果不足k到了链表尾,直接返回,不翻转
if (tail == NULL)
return head;
tail = tail->next;
}
//翻转时需要的前序和当前节点
ListNode* pre = NULL;
ListNode* cur = head;
//在到达当前段尾节点前
while (cur != tail) {
//翻转
ListNode* temp = cur->next;
cur->next = pre;
pre = cur;
cur = temp;
}
//当前尾指向下一段要翻转的链表
head->next = reverseKGroup(tail, k);
return pre;
}
};
学会此题,碰到反转链表类型题再也不怕了!