PAT(Python)-1025:反转链表（25）

给定一个常数K以及一个单链表L，请编写程序将L中每K个结点反转。例如：给定L为1→2→3→4→5→6，K为3，则输出应该为3→2→1→6→5→4；如果K为4，则输出应该为4→3→2→1→5→6，即最后不到K个元素不反转。

输入格式：

每个输入包含1个测试用例。每个测试用例第1行给出第1个结点的地址、结点总个数正整数N(<= 10⁵)、以及正整数K(<=N)，即要求反转的子链结点的个数。结点的地址是5位非负整数，NULL地址用-1表示。

接下来有N行，每行格式为：

Address Data Next

其中Address是结点地址，Data是该结点保存的整数数据，Next是下一结点的地址。

输出格式：

对每个测试用例，顺序输出反转后的链表，其上每个结点占一行，格式与输入相同。

实际提交代码发现测试点5始终过不去，折腾了一天也不知道到底还有什么问题没考虑到，不知道有没有人能给个提示或指出错误呢？

# -*- coding: utf-8 -*-
# 思路：从给定的链首到链末共有数据个数n加1项，id为前n项，next为后n项
def link(address,head,li):  # 采用递归将链表所用节点依次存入列表
    if head in address:
        arrow = address[head]
        li.append(arrow)
        return link(address,arrow,li)
    else:
        return li
head,N_all,cell = input().split()  # 读取第一行输入：首项元素id，元素个数N_all，反转间隔cell
N_all,cell = int(N_all),int(cell)  # 将N与cell转换为整数
address,data = {},{}  # 初始化两个字典，其中address保存id-next关系，data保存id-data关系
for i in range(N_all):  # 将输入信息存入address和data中
    a,b,c = input().split()
    address[a],data[a] = c,b   
# 调用link()函数将所给信息按链接顺序存入id_plus，有N_all+1个元素，末项为链尾指针，指向未知id或为-1
id_plus = link(address,head,[head])  # 第三项必须为[head]而不是[]，因为link()函数无法添加第一个id
id_plus[-1] = '-1'  # 指定链尾指针指向-1
N_e = len(id_plus) - 1  # id_plus中的元素个数为可链接到的元素，从首项到最后的-1，实际对应的数据个数应减1
num, rem = N_e // cell,