PAT 乙级练习 1090 危险品装箱

PAT 乙级练习 题解合集

本题链接

题目

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

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

输入格式：
输入第一行给出两个正整数：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

思路

注意：

• 一样物品可能和多样物品不相容，所以在记录不相容清单的时候要建立int到set<int>的映射。
• 每一次都要记录双向的数据，物品 a 和物品 b 不相容的同时说明物品 b 和物品 a 也不相容。

---

步骤：

1. map<int, set<int> > bombList;
   用bombList存储所有不相容的物品清单；
2. set<int> stuff;
   对每一箱货物的每一个物品input，在被放入stuff之前先要进行一次安全检查：stuff中是否存在和input不相容的物品？如果存在，则不安全。检查完之后将input放入stuff，继续检查下一个物品。

代码

#include <iostream>
#include <map>
#include <set>
using namespace std;
int main() {
	int input1, input2;
	int n, m, stuffCnt;
	cin >> n >> m;
	map<int, set<int> > bombList;
	set<int> stuff;
	while (n--) {
		cin >> input1 >> input2;
		bombList[input1].insert(input2);
		bombList[input2].insert(input1);
	}
	while (m--) {
		bool flag = true;
		cin >> stuffCnt;
		stuff.clear();
		for (int i = 0; i < stuffCnt; ++i){
			cin >> input1;
			for (auto it = bombList[input1].begin(); it != bombList[input1].end(); ++it)
				if (stuff.count(*it))
					flag = false;
			stuff.insert(input1);
		}
		cout << (flag ? "Yes" : "No") << endl; 
	} 
	return 0;
}