python小课——零基础入门——学习笔记刷题学编程Leetcode模拟面试

PTA 1090 危险品装箱 (25 分)

2021-09-06  本文已影响0人  freesan44

题目

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

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

输入格式:
输入第一行给出两个正整数: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
结尾无空行

解题思路

N, M = map(int, input().split())
# N, M = map(int, "6 3".split())
NDict = dict()#用字典Key+Arr方式保存冲突商品
for _ in range(N):
    N1, N2 = map(str, input().split())
    # N1, N2 = map(str, "10003 20004".split())
    if N1 not in NDict:
        NDict[N1] =[]
    if N2 not in NDict:
        NDict[N2] = []
    NDict[N1].append(N2)
    NDict[N2].append(N1)
for _ in range(M):
    MList = list(map(str, input().split()))[1:]
    # MList = list(map(str, "4 00001 20004 00002 20003".split()))[1:]
    isXiangtong = False
    for Meach in MList:
        if Meach in NDict:
            # print(Meach, NDict[Meach])
            arr = NDict[Meach]
            for i in arr:
                if i in MList:
                    isXiangtong = True
                    break
        if isXiangtong == True:break#多次跳出循环减少耗时
    if isXiangtong == True:
        print("No")
    else:
        print("Yes")
上一篇 下一篇

猜你喜欢

热点阅读