leetcode--25. 每k个一组翻转链表

题目：25. 每k个一组翻转链表

链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group/description/

给定一个链表，每k个节点进行翻转，对于末尾不足k个节点的部分要保持原状。要求空间复杂度O(1)。比如对于链表1->2->3->4->5，当k=2的时候，返回2->1->4->3->5，k=3的时候返回3->2->1->4->5。

先写了一个翻转整个链表的函数，返回翻转后的头节点和尾节点。然后把原链表k个一组拆开，分别翻转拼接。

python：

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

class Solution:
    def reverseKGroup(self, head, k):
        """
        :type head: ListNode
        :type k: int
        :rtype: ListNode
        """
        if not head or not head.next or k==1:
            return head
        def revertAll(head):
            if not head or not head.next:
                return head
            p1, p2, p3 = head, head.next, head.next.next
            p1.next,ret = None,list()
            ret.append(head)
            while p2:
                p2.next, p1, p2 = p1, p2, p3
                if not p2:
                    ret.append(p1)
                    break
                p3 = p3.next
            return ret
        retHead=temp=ListNode(-1)
        p1=p2=p3=head
        while p1:
            cnt,p2=1,p1
            while cnt<k:
                if p2:
                    p2=p2.next
                else:
                    break
                cnt+=1
            if not p2:
                temp.next=p1
                break
            p3=p2.next
            p2.next=None
            ret=revertAll(p1)
            temp.next=ret[1]
            temp=ret[0]
            p1=p3
        return retHead.next