pat 乙级 1090 危险品装箱

Posted 2021-01-19 eat-too-much

题目

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

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

输入第一行给出两个正整数：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 <map>
#include <vector>
#include <algorithm>
using namespace std;

int main(){
    int N,M;
    map<int,vector<int> > mp;
    cin >> N >> M;
    for(int i=0; i< N; i++){
        int t1 = 0;
        int t2 = 0;
        cin >> t1 >> t2;
        mp[t1].push_back(t2);
        mp[t2].push_back(t1);
    }
    for(int i=0; i<M; i++){
        int k=0;
        bool flag = true;
        int temp = 0;
        cin >> k;
        vector<int> v;
        for(int j = 0; j< k; j++){
            cin >> temp;
            v.push_back(temp);
        }
        for(int j=0; j<v.size(); j++){
            for(int t=0; t<mp[v[j]].size(); t++){
                if(find( v.begin(), v.end() , mp[v[j]][t]) != v.end()){
                    flag = false;
                } 
            }
        }
        if(flag){
            cout << "Yes" << endl;
        }else{
            cout << "No" << endl;
        }
    }
}