PAT乙级 1090危险品装箱 C++

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

注意每种货物可能与多个物品不相容，存储每种货物的不相容货物，在货物清单中寻找它们，注意遍历容量小的容器，本题货物清单容器容量小，不然可能会超时

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <map>
#include <vector>

using namespace std;

int main() {
    int n, m;
    cin >> n >> m;
    map<string, vector<string>> wrong;//每种货物对应一个与它不相容的货物容器
    for(int i = 1; i <= n; i++) {
        string str1, str2;
        cin >> str1 >> str2;
        wrong[str1].push_back(str2);
        wrong[str2].push_back(str1);
    }
    int result[m] = {0}; //存储检查结果，1可以，-1不可以
    for(int i = 1; i <= m; i++) {
        int sum;
        cin >> sum;
        vector<string> foods;//本次的货物清单
        int flag = 1;
        for(int a = 1; a <= sum; a++) {
            string str;
            cin >> str;
            foods.push_back(str);
        }
        for(int a = 1; a <= sum; a++) {
            if(wrong[foods.at(a - 1)].size() != 0) {//如果该货物存在不相容物品，进入检查
                vector<string> thewrong = wrong[foods.at(a - 1)];
                for(int b = 1; b <= sum; b++) {//遍历货物清单寻找不相容物品
                    vector<string>::iterator it2 = find(thewrong.begin(), thewrong.end(), foods.at(b - 1));
                    if(it2 != thewrong.end()) {//如果清单存在不相容物品
                        flag = -1;
                        goto wancheng;//来到大循环最后
                    }
                }
            }
        }
wancheng:
        result[i - 1] = flag;
    }
    for(int i = 1; i <= m; i++) {
        if(result[i - 1] == -1)
            cout << "No" << endl;
        else
            cout << "Yes" << endl;
    }
    return 0;
}