go语言刷题：25. K 个一组翻转链表

申明：本文只用做自己学习记录

题目

给定一个链表，每 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]

代码

func reverseKGroup(head *ListNode, k int) *ListNode {
	//创建链表头
	dummy := &ListNode{Next: head}
	//创建指针
	temp := dummy
	//当链表不为空时
	for head != nil {
		tail := temp
		for i := 0; i < k; i++ {
			//将尾部指针移动到k的位置，如果第k个位置链表为空则直接返回整个链表
			tail = tail.Next
			if tail == nil {
				return dummy.Next
			}
		}
		//将k+1个节点提取出来
		nex := tail.Next
		//将前k个节点倒置
		head, tail = myReverse(head, tail)
		//将头尾连接上
		temp.Next = head
		tail.Next = nex

		//更新指针
		temp = tail
		head = tail.Next
	}
	return dummy.Next
}

func myReverse(head, tail *ListNode) (*ListNode, *ListNode) {
	//取出尾部指针
	Tail := tail.Next
	//取出头部指针
	Head := head
	for Tail != tail {
		//取出头部指针的下一节点指针
		temp := Head.Next
		//将头部指针下一节点指针指向尾结点
		Head.Next = Tail
		//将头节点变为尾部节点
		Tail = Head
		//更新头节点
		Head = temp
	}
	return tail, head
}