PAT 乙级 1090.危险品装箱 C++/Java
集装箱运输货物时，我们必须特别小心，不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱，否则很容易造成爆炸。
本题给定一张不相容物品的清单，需要你检查每一张集装箱货品清单，判断它们是否能装在同一只箱子里。
输入格式：
输入第一行给出两个正整数：N (≤104) 是成对的不相容物品的对数；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
分析：
输入N，M，分别为：
- N：不相容物品的对数
- M：物品清单单数
接下来N行，每行输入两个数，表示不相容的物品
接下来M行，每行给出一箱货物的清单
思路：
一开始用了unordered_map<int, unordered_map<int, int>> m来存放，超内存了
后来考虑用一个map和vector数组来存放不相容的物品
比如下面这个输入，我们可以用unordered_map<int, vector<int>> m来存放
把第一第二个物品都放入map中，记录不相容的物品
20001 20002
20003 20004
20005 20006
20003 20001
20005 20004
20004 20006
这样就知道了每个物品和哪些东西不相容：
key(int) -> value(vector<int>)
m[20001] -> 20002, 20003
m[20002] -> 20001
m[20003] -> 20001, 20004
m[20004] -> 20003, 20005, 20006
m[20005] -> 20004, 20006
m[20006] -> 20004, 20005
对于接下来的M行输入，每行的输入存入一个数组res
4 00001 20004 00002 20003
5 98823 20002 20003 20006 10010
3 12345 67890 23333
比如第一行的4 00001 20004 00002 20003
存入数组res得到：00001 20004 00002 20003
同时用一个数组arr记录这4个id的出现情况
比如（用下标记录）：
arr[00001] = 1
arr[20004] = 1
arr[00002] = 1
arr[20003] = 1
遍历数组res，在unordered_map中，获取到与res[i]不相容的物品
m[res[i]]就是不相容的物品，这是一个vector数组
- 比如
res[i] = 20001：m[res[i]] = m[20001] -> 20002, 20003
遍历这个数组，取出里面的元素x
- 如果
arr[x] == 1：就说明有不相容的物品
C++实现：
#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_map>
#include <unordered_set>
#include <string>
#include <stack>
#include <cmath>
#include <map>
using namespace std;
int main() {
short N; // 成对的，不相容物品的对数
short M; // 集装箱货品清单的单数
unordered_map<int, vector<int>>m;
cin >> N >> M;
for (int i = 0; i < N; ++i) {
int first, second;
cin >> first >> second;
m[first].push_back(second);
m[second].push_back(first);
}
while (M--) {
bool flag = false;
int cnt;
cin >> cnt;
vector<int> res(cnt);
int arr[100001] = { 0 };
for (int i = 0; i < cnt; ++i) {
cin >> res[i];
arr[res[i]] = 1;
}
for (int i = 0; i < cnt; ++i) {
for (int j = 0; j < m[res[i]].size(); ++j) {
// m[res[i]]是一个vector数组
// m[res[i]][j]是vecotr里的元素
int x = m[res[i]][j];
if (arr[x] == 1) {
flag = true;
}
}
}
if (flag) {
printf("No\n");
}
else {
printf("Yes\n");
}
}
return 0;
}
Java实现：
// TODO
