给你链表的头节点 head ，每 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]
提示：
链表中的节点数目为 n
1 <= k <= n <= 5000
0 <= Node.val <= 1000
**进阶：**你可以设计一个只用 O(1) 额外内存空间的算法解决此问题吗？
模拟，每次获取要反转的头节点和尾结点
反转完成后返回新的头节点和尾结点
然后把反转结果接入
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
| Bash class Solution {
public:
ListNode *reverseKGroup(ListNode *head, int k) {
ListNode *pre = new ListNode(-1, head);
ListNode *ans = pre;
while (head) {
ListNode *tail = pre;
for (int i = 0; i < k; i++) {
tail = tail->next;
if (tail == nullptr)
return ans->next;
}
ListNode *mid = tail->next;
auto g = reverseList(head, tail);
g.second->next = mid;
pre->next = g.first;
pre = g.second;
head = mid;
}
return ans->next;
}
pair<ListNode *, ListNode *> reverseList(ListNode *head, ListNode *tail) {
ListNode *pre = nullptr;
ListNode *cur = head;
while (pre != tail) {
ListNode *mid = cur->next;
cur->next = pre;
pre = cur;
cur = mid;
}
return {tail, head};
}
};
|
