25. K 个一组翻转链表

给你链表的头节点 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
35
36
37
38
39
40
41
42
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func reverseKGroup(head *ListNode, k int) *ListNode {
poi := &ListNode{Next: head}
pre := poi

for head != nil {
tail := pre
for i:=0; i<k; i++ {
tail = tail.Next
if tail == nil {
return poi.Next
}
}
next := tail.Next
head, tail = reverse(head, tail)

pre.Next = head
tail.Next = next
pre = tail
head = tail.Next
}

return poi.Next
}

func reverse(start, end *ListNode) (*ListNode, *ListNode) {
pre := end.Next
cur := start
for pre != end {
next := cur.Next
cur.Next = pre
pre = cur
cur = next
}
return end, start
}