L2-002 链表去重分数

给定一个带整数键值的链表 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

思路参照：http://t.csdnimg.cn/utbVA 

 不超时的思路：除了直接从标准输入流读取、输出数据还可以用集合降重，然后省掉中间列表，减少append操作。然后就能过啦

import sys

def pr(nodes):
    for i, node in enumerate(nodes):
        if i == len(nodes) - 1:
            sys.stdout.write("%s %s -1\n" % (node[0], node[1]))
        else:
            sys.stdout.write("%s %s %s\n" % (node[0], node[1], nodes[i + 1][0]))

address, n = sys.stdin.readline().split()
nodes = {}
unique_keys = set()
good_nodes = []
bad_nodes = []

for _ in range(int(n)):
    a, b, c = sys.stdin.readline().split()
    nodes[a] = (a, b, c)
    unique_keys.add(abs(int(b)))

current_address = address
while current_address != '-1':
    current_node = nodes[current_address]
    if abs(int(current_node[1])) in unique_keys:
        good_nodes.append(current_node)
        unique_keys.remove(abs(int(current_node[1])))
    else:
        bad_nodes.append(current_node)
    current_address = current_node[2]

pr(good_nodes)
if bad_nodes:
    pr(bad_nodes)