PAT 1090 危险品装箱

https://pintia.cn/problem-sets/994805260223102976/problems/1038429484026175488

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

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

输入格式：

输入第一行给出两个正整数：N (≤) 是成对的不相容物品的对数；M (≤) 是集装箱货品清单的单数。

随后数据分两大块给出。第一块有 N 行，每行给出一对不相容的物品。第二块有 M 行，每行给出一箱货物的清单，格式如下：

K G[1] G[2] ... G[K]

其中 K (≤) 是物品件数，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 <bits/stdc++.h>

using namespace std;

const int maxn = 1e5 + 10;

int N, M;

map<int, vector<int> > mp;

int vis[maxn];

int main() {

    scanf("%d%d", &N, &M);

    for(int i = 0; i < N; i ++) {

        int x, y;

        scanf("%d%d", &x, &y);

        mp[x].push_back(y);

        mp[y].push_back(x);

    }

    while(M --) {

        int ans;

        scanf("%d", &ans);

        memset(vis, 0, sizeof(vis));

        vector<int> v(ans);

        for(int i = 0; i < ans; i ++) {

            scanf("%d", &v[i]);

            vis[v[i]] = 1;

        }

        bool flag = true;

        for(int i = 0; i < v.size(); i ++) {

            for(int j = 0; j < mp[v[i]].size(); j ++)

                if(vis[mp[v[i]][j]]) flag = false;

        }

        if(flag) printf("Yes\n");

        else printf("No\n");

    }

    return 0;

}