leetcode k个一组翻转链表

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

k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。
leetcode k个一组翻转链表


与上一个博客一样,采用递归的思想,这里注意需要保存newHeadNext,否则会产生循环,造成超时。

class Solution:
    def reverseKGroup(self, head, k):
        """
        :type head: ListNode
        :type k: int
        :rtype: ListNode
        """
        if head == None:
            return None
        listNode = []
        p = head
        for i in range(k):
            listNode.append(p)
            p = p.next
            if i < k - 1 and p == None:
                return head
        newHead = listNode[k - 1]
        newHeadNext = self.reverseKGroup(newHead.next, k)
        for i in range(k - 1, 0, -1):
            listNode[i].next = listNode[i - 1]
        listNode[0].next = newHeadNext
        return newHead

leetcode k个一组翻转链表