25. K 个一组翻转链表

题目

给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。

k 是一个正整数,它的值小于或等于链表的长度。

如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。

进阶:

你可以设计一个只使用常数额外空间的算法来解决此问题吗?
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。

示例 1:

img
1
2
输入:head = [1,2,3,4,5], k = 2
输出:[2,1,4,3,5]

示例 2:

img
1
2
输入:head = [1,2,3,4,5], k = 3
输出:[3,2,1,4,5]

示例 3:

1
2
输入:head = [1,2,3,4,5], k = 1
输出:[1,2,3,4,5]

示例 4:

1
2
输入:head = [1], k = 1
输出:[1]
1
2
3
4
5
6
提示:

列表中节点的数量在范围 sz 内
1 <= sz <= 5000
0 <= Node.val <= 1000
1 <= k <= sz

思路

每k个节点进行一次划分,这个内部是一个翻转链表,我们需要记录链表的start和end,如果最后的不足k个,我们需要把end当做头部,start当做尾部

实现

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
35
36
37
38
39
40
41
42
43
44
45
46
class Solution {
public ListNode reverseKGroup(ListNode head, int k) {
// 12345
// 012345
// 返回的头部节点
ListNode dummy = new ListNode(0);
dummy.next = head;
// 需要翻转的节点的头部的上一个节点
ListNode pre = dummy;
// 翻转链表后的末尾
ListNode end = dummy;
// 021345
while (end.next != null) {
for (int i = 0; i < k && end != null; i++) {
end = end.next;
}
// 说明不足一个k,不需要翻转
if (end == null) {
break;
}

ListNode start = pre.next;
ListNode next = end.next;

end.next = null;
pre.next = reserve(start);
start.next = next;
pre = start;
end = pre;
}
return dummy.next;

}

private ListNode reserve(ListNode head) {
ListNode pre = null;
ListNode curr = head;
while (curr != null) {
ListNode temp = curr.next;
curr.next = pre;
pre = curr;
curr = temp;
}
return pre;
}
}