给定一个带整数键值的链表 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
不超时的思路:除了直接从标准输入流读取、输出数据还可以用集合降重,然后省掉中间列表,减少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)