PAT乙级真题 1090 危险品装箱 C++实现

题目

集装箱运输货物时，我们必须特别小心，不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱，否则很容易造成爆炸。
本题给定一张不相容物品的清单，需要你检查每一张集装箱货品清单，判断它们是否能装在同一只箱子里。
输入格式：
输入第一行给出两个正整数：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

思路

这道题25分，却比上道20分的狼人杀-简单版容易多了。

不相容的物品是多对多关系，可用二维vector记录不相容的编号对，即每个编号对应一个与其不相容的vector，如v[20001] = {20002, 20003}。

对于一箱中的每件货物，如果其不相容数组里有记录，则在后面的元素中查找是否存在不相容编号（只需遍历其后面的，因为不相容关系是互相关联的）。

代码

#include <iostream>
#include <vector>
using namespace std;

int main(){
    int n, m;
    cin >> n >> m;
    vector<vector<int> > v(100000);
    for (int i=0; i<n; i++){
        int g1, g2;
        cin >> g1 >> g2;
        v[g1].push_back(g2);
        v[g2].push_back(g1);
    }
    for (int i=0; i<m; i++){
        int k;
        cin >> k;
        vector<int> g(k);
        for (int j=0; j<k; j++){
            cin >> g[j];
        }
        bool isSafe = true;
        for (int j=0; j<k-1 && isSafe; j++){ //遍历到倒数第2个即可
            for (int l=0; l<v[g[j]].size() && isSafe; l++){
                for (int m=j+1; m<k; m++){
                    if (v[g[j]][l]==g[m]){
                        isSafe = false;
                        break;
                    }
                }
            }
        }
        if (isSafe){
            cout << "Yes";
        }
        else {
            cout << "No";
        }
        cout << endl;
    }
    return 0;
}