python-PAT-1090 危险品装箱-超时问题

输入格式：
输入第一行给出两个正整数：N (≤10?4??) 是成对的不相容物品的对数；M (≤100) 是集装箱货品清单的单数。
随后数据分两大块给出。第一块有 N 行，每行给出一对不相容的物品。第二块有 M 行，每行给出一箱货物的清单，格式如下：
K G[1] G[2] … G[K]
其中 K (≤1000) 是物品件数，G[i] 是物品的编号。简单起见，每件物品用一个 5 位数的编号代表。两个数字之间用空格分隔。

输出格式：
对每箱货物清单，判断是否可以安全运输。如果没有不相容物品，则在一行中输出 Yes，否则输出 No。
输入样例：
6 3
20001 20002
20003 20004
20005 20006
20003 20001
20005 20004
20004 20006
4 00001 20004 00002 20003
5 98823 20002 20003 20006 10010
3 12345 67890 23333

输出样例：
No
Yes
Yes

后两个测试点超时版本

info1 = input().split()
n = int(info1[0])
m = int(info1[1])
rules = set()
for i in range(n):info2 = input().split()rules.add((info2[0], info2[1]))rules.add((info2[1], info2[0]))
for i in range(m):info3 = input().split()t = int(info3[0])info4 = info3[1:]flag = Falsefor a in range(t - 1):for b in range(a + 1, t):if (info4[a], info4[b]) in rules or (info4[b], info4[a]) in rules:flag = Truebreakif flag:breakif flag:print('No')else:print('Yes')

更改后的

info1 = input().split()
n = int(info1[0])
m = int(info1[1])
rules = {
    }
for i in range(n):info2 = input().split()rules[info2[0]] = rules.get(info2[0],[]) + [info2[1]]
for i in range(m):info3 = input().split()t = int(info3[0])info4 = info3[1:]flag = Falsefor a in info4:if a in rules:b = rules[a]for j in b:if j in info4:flag = Truebreakif flag:breakif flag:print('No')else:print('Yes')

解释：版本1按对遍历太耗时了。因为危险物品是成对出现的，那如果某id已经在危险名录了，那么检查它的不相容对象是不是也在运单上，能更快判断非法情况。