PAT (Basic Level) Practice （中文）1090 危险品装箱 (25 分)

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

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

输入格式

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

随后数据分两大块给出。第一块有 N N N行，每行给出一对不相容的物品。第二块有 M M 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

思路

• 利用哈希表储存不相容物品, 因为有的物品和多个物品不相容, 使用unordered_map<int,vector<int> >
• 每次检验每箱物品的时候, 遍历每一个物品,找出和这个物品不相容的其他物品们是不是存在于这箱物品之中

AC code

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

unordered_map<int,vector<int> > mp;
bool judge(){
    int n;
    scanf("%d",&n);
    unordered_map<int,bool> items;
    vector<int> v(n);
    for(int i=0;i<n;i++){
        scanf("%d",&v[i]);
        items[v[i]]=true;
    }
    for(int i=0;i<n;i++){   //遍历所有的物品
        for(auto& x:mp[v[i]]){   //遍历和这个物品不相容的物品列表
            if(items[x]) return false;  //如果有不相容的物品存在于这行物体中
        }
    }
    return true;
}
int main(){
    int N,M,a,b;
    scanf("%d %d",&N,&M);
    for(int i=0;i<N;i++){
        scanf("%d %d",&a,&b);
        mp[a].push_back(b);
        mp[b].push_back(a);
    }    
    for(int i=0;i<M;i++){
        if(judge()) cout<<"Yes" <<endl;
        else cout<<"No" << endl;
    }
    return 0;
}