L2-002 链表去重(Python3)

给定一个带整数键值的链表 L，你需要把其中绝对值重复的键值结点删掉。即对每个键值 K，只有第一个绝对值等于 K 的结点被保留。同时，所有被删除的结点须被保存在另一个链表上。例如给定 L 为 21→-15→-15→-7→15，你需要输出去重后的链表 21→-15→-7，还有被删除的链表 -15→15。

输入格式：

输入在第一行给出 L 的第一个结点的地址和一个正整数 N（≤105，为结点总数）。一个结点的地址是非负的 5 位整数，空地址 NULL 用 −1 来表示。

随后 N 行，每行按以下格式描述一个结点：

地址 键值 下一个结点

其中地址是该结点的地址，键值是绝对值不超过104的整数，下一个结点是下个结点的地址。

输出格式：

首先输出去重后的链表，然后输出被删除的链表。每个结点占一行，按输入的格式输出。

输入样例：

00100 5
99999 -7 87654
23854 -15 00000
87654 15 -1
00000 -15 99999
00100 21 23854

输出样例：

00100 21 23854
23854 -15 99999
99999 -7 -1
00000 -15 87654
87654 15 -1

提交结果（如果运行超时多试几次）：

代码：

import sys


def pd(a):
    for i in range(len(a)):
        if i == len(a) - 1:
            print("{} {} -1".format(a[i][0], a[i][1]))
            return
        print("{} {} {}".format(a[i][0], a[i][1], a[i + 1][0]))


L, n = sys.stdin.readline().split()
v = []
good = []
bad = []
num = {}
node = {}
for i in range(int(n)):
    a, b, c = sys.stdin.readline().split()
    num[abs(int(b))] = 0
    node[a] = []
    node[a].append(a)
    node[a].append(b)
    node[a].append(c)
while L != "-1":
    v.append(node[L])
    L = node[L][-1]
for i in v:
    if num[abs(int(i[1]))] == 0:
        good.append(i)
        num[abs(int(i[1]))] = 1
    else:
        bad.append(i)
pd(good)
if len(bad) > 0:
    pd(bad)