Python基础练习 1090 危险品装箱 (25 分)

本文已参与「新人创作礼」活动，一起开启掘金创作之路。

问题描述

　　集装箱运输货物时，我们必须特别小心，不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱，否则很容易造成爆炸。本题给定一张不相容物品的清单，需要你检查每一张集装箱货品清单，判断它们是否能装在同一只箱子里。

输入输出

• 输入格式

　　输入第一行给出两个正整数：N (≤10000​​ ) 是成对的不相容物品的对数；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

Python版本 代码如下

本题没有出现用append()函数追加超时的情况，因此选用了append()函数，但实际上更为推荐使用extend()函数。
思路：主要记得用字典来解决。

n, m = input().split()

dit={}  # 追加到字典中
for i in range(int(n)):
    x = input().split()
    if x[0] not in dit:
        dit[x[0]] = []
    if x[1] not in dit:
        dit[x[1]] = []
    dit[x[0]].append(x[1])  # 注意函数的选择
    dit[x[1]].append(x[0])  # 注意函数的选择

for i in range(int(m)):
    flag=1
    s = input().split()
    for e in s:
        if e in dit:
            for k in dit[e]:
                if k in s:
                    flag=0
                    print("No")
                    break
        if flag==0:
            break

    if flag==1:
        print("Yes")