leetcode-25. K 个一组翻转链表

题目

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

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

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

示例：

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

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

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

说明：

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

解题思路

头条的面试题……可惜当时没写出来

先考虑反转某段链表（用3个指针分别保存前一个节点，当前节点，当前节点的后一个节点，反转后要保证不能断链），然后再对整段链表进行反转。见反转链表 II

核心是返回反转后的new_head和new_end

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:

        def reverse_list(head, end) -> tuple:
            end_next = end.next
            pre, p = end_next, head
            while p != end_next:
                pn = p.next
                p.next = pre
                pre, p = p, pn
            return (end, head)

        ret_head = ListNode(-1)
        ret_head.next = head
        reverse_pre, reverse_end = ret_head, ret_head
        cnter = 0
        while reverse_end:
            if cnter == k:
                new_head, new_end = reverse_list(reverse_pre.next, reverse_end)
                reverse_pre.next = new_head
                reverse_pre = new_end
                reverse_end = reverse_pre
                cnter = 0
            reverse_end = reverse_end.next
            cnter += 1
        return ret_head.next