PAT 乙级1090危险品装箱

题目:

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

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

输入格式:

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

代码: 

#include
using namespace std;

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

    map>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;
        setwupin;
        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;
}

你可能感兴趣的:(c++,算法,c++)