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

25. K 个一组翻转链表

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

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

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

示例：

给你这个链表：1->2->3->4->5

当 k = 2 时，应当返回: 2->1->4->3->5

当 k = 3 时，应当返回: 3->2->1->4->5

说明:

• 你的算法只能使用常数的额外空间。
• 你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

答案：

type ListNode struct {
	Val  int
	Next *ListNode
}

func reverseKGroup(head *ListNode, k int) *ListNode {
	if k <= 1 || head == nil || head.Next == nil {
		return head
	}

	var result *ListNode
	var rListNode *ListNode
	count := 0
	for head != nil {
		cur := head
		next := head.Next
		//处理当前节点
		if rListNode == nil {
			rListNode = cur
			rListNode.Next = nil
		} else {
			cur.Next = rListNode
			rListNode = cur
		}
		head = next

		//计数
		count++
		if count == k {
			result = addToLast(result, rListNode)
			//清空计数和反转链表
			count = 0
			rListNode = nil
		}
	}

	//处理非整数倍数据
	if count != 0 {
		result = addToLast(result, reverseKGroup(rListNode, count))
	}

	return result
}

//加至末尾
func addToLast(head *ListNode, add *ListNode) *ListNode {
	if head == nil {
		head = add
	} else {
		tListNode := head
		for tListNode.Next != nil {
			tListNode = tListNode.Next
		}
		tListNode.Next = add
	}
	return head
}