题目
给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。
进阶:
你可以设计一个只使用常数额外空间的算法来解决此问题吗?
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
示例 1:
输入:head = [1,2,3,4,5], k = 2
输出:[2,1,4,3,5]
示例 2:
输入:head = [1,2,3,4,5], k = 3
输出:[3,2,1,4,5]
示例 3:
输入:head = [1,2,3,4,5], k = 1
输出:[1,2,3,4,5]
示例 4:
输入:head = [1], k = 1
输出:[1]
提示:
列表中节点的数量在范围 sz 内
1 <= sz <= 5000
0 <= Node.val <= 1000
1 <= k <= sz
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-nodes-in-k-group
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
写起来比较复杂的模拟。我的核心思路是构造函数 reverseLinklist
,举例说明这个函数的使用方法:
对于 输入:head = [1,2,3,4,5], k = 2
,当一次翻转之后链表站长这样:head = [2,1,3,4,5]
,此时要把结点 3、4 变成 4、3,reverseLinklist
会接收结点 3、4 的指针的引用,然后在函数内部用头插法把要反转的范围逆序构造成新的链表,再返回结点 4、结点 3 的指针。返回之后在原来的链表里面,结点 1 要和结点 4 接上,结点 3 要和结点 5 接上,也就是变成这样:head = [2,1,4,3,5]
有了这个函数之后,在遍历链表的时候需要额外的两个指针 prev
和 next
,在上面那个例子里面 prev
和 next
就分别指向结点 1 和结点 5,prev
和 next
中间就是要反转的部分,反转结束之后再和 prev
和 next
接上,然后继续往后跑到下一个要反转的地方。
给原来的链表接上一个空的头节点会更方便一点(官方代码的彩蛋,head
结点再接上一个空的头节点,就是 hair
结点了,合情合理,hhh)。
时间复杂度:
O
(
n
)
O(n)
O(n)
空间复杂度:
O
(
n
)
O(n)
O(n)
C++ 代码
/**
* 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 (k == 1 || !head)
return head;
ListNode* myHead = new ListNode(0);
myHead->next = head;
ListNode* prev = myHead, *next;
ListNode* p = head, *q = head;
for (int i = 0; i < k - 1; ++i) {
if (!q->next) {
ListNode* ret = myHead->next;
delete myHead;
return ret;
}
q = q->next;
}
next = q->next;
while (true) {
reverseLinklist(p, q);
prev->next = p;
q->next = next;
for (int i = 0; i < k; ++i) {
if (!q->next) {
ListNode* ret = myHead->next;
delete myHead;
return ret;
}
p = p->next;
q = q->next;
prev = prev->next;
}
next = q->next;
}
return head;
}
void reverseLinklist(ListNode*& head, ListNode*& tail) {
if (head == tail)
return;
ListNode* newHead = new ListNode(0);
ListNode* newTail = head;
ListNode* p = head;
tail->next = nullptr;
while (p) {
ListNode* q = p->next;
p->next = newHead->next;
newHead->next = p;
p = q;
}
head = newHead->next;
tail = newTail;
delete newHead;
}
};