PAT 乙级1090危险品装箱

good luck for my 192023-10-28 17:11

题目：

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

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

输入格式：

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

代码：

cpp 复制代码

#include<bits/stdc++.h>
using namespace std;

int main() {
    int n, m;
    cin >> n >> m;

    map<int, vector<int>>mp;
    for (int i = 0; i < n; i++) {
        int not1, not2;
        cin >> not1 >> not2;
        mp[not1].push_back(not2);
        mp[not2].push_back(not1);
    }
    int k;

    for (int j = 0; j < m; j++) {
        cin >> k;
        set<int>wupin;
        for (int i = 0; i < k; i++) {
            int thing;
            cin >> thing;
            wupin.insert(thing);
        }
        bool flag = false; // 重置标志变量
        bool flag1=false;
        for (auto it = wupin.begin(); it != wupin.end(); it++) {
            int a = *it;
            if (mp.find(a) != mp.end()) {
                for(auto itt = mp[a].begin(); itt != mp[a].end(); itt++){
                    if(wupin.find(*itt)!=wupin.end()){
                        flag = true;
                        if(!flag1){
                            cout << "No" << endl;
                        }
                        
                        flag1=true;
                        break;
                    }    
                }
            
            }
            else continue;
        }
        if (!flag) cout << "Yes" << endl;
        wupin.clear();
    }
    return 0;
}