PAT 甲级 1052 Linked List Sorting

1052 Linked List Sorting

题目大意：以（address,key,next）形式给出一个链表，将链表按key进行排序后输出新生成的链表

思路：本以为直接排序就好，next给的数据都没有用，后来发现会有多的节点，即给出一个头节点会生成一个链表，但是有的节点不在这条链上，这就要去掉多余的节点。测试点3超时，C++完美AC。想了一下输入输出是不能砍的，唯一能动手脚的应该就是去除节点的步骤了，不过性价比不高，太费时间

n,head=map(int,input().split())
l=[]
l1=[]
index={}    #记录每个地址所在的下标
for i in range(n):
    l.append(tuple(map(int,input().split())))
    index[l[i][0]]=i
start=head
if start not in index:  #不在已存在的下标中
    print("0 -1")
else:
    while l[index[start]][2]!=-1:   #该步骤是去除掉不在链表里的元素
        l1.append(l[index[start]])
        start=l[index[start]][2]
    l1.append(l[index[start]])
    l1.sort(key=lambda x:x[1])
    print("{:d} {:05d}".format(len(l1),l1[0][0]))
    for i in range(len(l1)):
        if i!=len(l1)-1:
            print("{:05d} {:d} {:05d}".format(l1[i][0],l1[i][1],l1[i+1][0]))
        else:
            print("{:05d} {:d} -1".format(l1[i][0],l1[i][1]))