Bootstrap

[Python][PAT]1090 危险品装箱 (25 分)

1090 危险品装箱 (25 分)

集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸。

本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否能装在同一只箱子里。

输入格式:

输入第一行给出两个正整数: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

思路:

主要是用python的字典,组成一个个键值对,,当然有些key 可能存在多个value(比如样例数据中的20003,它对应20001,20004的时候都不能进行存放),所以代码中采用了一个键对上多个值的方式,把值当成字符串,以空格分割

代码:

def main():
    n = input()
    n = n.split()
    Check = {}
    flag = True
    for i in range(int(n[0])):
        temp = input()
        temp = temp.split()
        if temp[0] in Check:
            Check[temp[0]] = Check[temp[0]]+" "+temp[1]
        else:
            Check[temp[0]] = temp[1]
        if temp[1] in Check:
            Check[temp[1]] = Check[temp[1]]+" "+temp[0]
        else:
            Check[temp[1]] = temp[0]
    for i in range(int(n[1])):
        temp = input()
        temp = temp.split()
        for j in range(1, len(temp)):
            if temp[j] not in Check:
                continue
            temp2 = Check[temp[j]]
            temp2 = temp2.split()
            for k in range(1, len(temp)):
                if temp[k] in temp2:
                    flag = False
                    break
            if not flag:
                print("No")
                break
        if flag:
            print("Yes")
        else:
            flag = True


main()

 

;