1090 危险品装箱(25) - PAT乙级

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

Solution:

将每箱货物的存成hash表, 遍历每对违禁物品, 两件物品都存在这张表里就说明要爆炸, 退出遍历.

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

struct Node {
string a, b;
};

int main() {
int N, M; cin >> N >> M;
vector<Node> v(N);
for (int i = 0; i < N; i++) {
cin >> v[i].a >> v[i].b;
}
unordered_map<string, int> mp;
int n; string temp;
while (M--) {
cin >> n; mp.clear();
for (int i = 0; i < n; i++) {
cin >> temp;
mp[temp] = 1;
}
int ok = 1;
for (int i = 0; i < N; i++) {
if (mp.count(v[i].a) && mp.count(v[i].b)) {
ok = 0; break;
}
}
if (ok) {
cout << "Yes" << endl;
} else {
cout << "No" << endl;
}
}
return 0;
}