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