[Leetcode 25] Reverse Nodes in k Group

来源:互联网 发布:python培训机构 编辑:程序博客网 时间:2024/06/08 06:03
# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None


class Solution(object):
    def reverseKGroup(self, head, k):
        """
        :type head: ListNode
        :type k: int
        :rtype: ListNode
        """
        if k==1 :
            return head
        m=self.getSize(head)
        dummy=cur=prev=ListNode(0)
        dummy.next=head
        n=m/k
        print k,m
        for i in range(n):
            for j in range(k):
                tmp=head.next
                head.next=prev
                prev=head
                head=tmp
            oldhead=cur.next
            cur.next=prev
            prev=oldhead
            prev.next=head
            cur=prev
        return dummy.next
    def getSize(self,head):
        n=0
        while head:
            n += 1
            head = head.next
        return n
0 0