每k个一组反转链表-python版

题目：传送门

给出一个链表，每 k 个节点一组进行翻转，并返回翻转后的链表。k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么将最后剩余节点保持原有顺序。

说明：

1. 你需要自行定义链表结构，将输入的数据保存到你的链表中；
2. 你不能只是单纯的改变节点内部的值，而是需要实际的进行节点交换；
3. 你的算法只能使用常数的额外空间。
   输入描述:
   第一行输入是链表的值
   第二行输入是K的值，K是大于或等于1的整数

输入形式为：
1 2 3 4 5
2
输出描述:
当 k = 2 时，应当输出:
2 1 4 3 5

当 k = 3 时，应当输出:
3 2 1 4 5

当k=6时，应当输出：
1 2 3 4 5
示例1
输入
1 2 3 4 5
2
输出
2 1 4 3 5

代码

class ListNode:
    def  __init__(self,x):
        self.val = x
        self.next = None

class Main:
    def const(self,arr):
        curr = ans = ListNode(-1)
        for x in arr:
            curr.next = ListNode(x)
            curr = curr.next
        return ans.next

    def reverse(self,listnode,n):
        ans = curr = ListNode(-1)
        while listnode:
            t = n
            arr = []
            while t >= 1 and listnode:
                arr.append(listnode.val)
                t -= 1
                listnode = listnode.next
            if t == 0:
                curr.next = self.const(reversed(arr))
            else:
                curr.next = self.const(arr)#如果最后一组不够n个，不需要反转
                
            tt = n #指针后移n位
            while tt >= 1:
                curr = curr.next
                tt -= 1
        ans = ans.next
        while ans:
            print(ans.val,end=' ')
            ans = ans.next

if __name__ == '__main__':
    arr = input()
    n = input()
    ln = Main().const(arr.split(' '))
    Main().reverse(ln, int(n))

注意事项

1. python读取控制台输入：arr = input()获取一行
2. python不换行输出：print(xxx,end=’ ')
3. 其他的看代码注释