集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸。
本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否能装在同一只箱子里。
输入格式:
输入第一行给出两个正整数: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
个人思路
这题在考试的时候我没做出来,今天写的时候对STL的各种工具有了进一步了解,所以本题主要是使用了map和set容器解决的。
每个有不相容的货物编号都映射一个set容器,内部存放所有与之不相容的货物的编号,而这个映射的关系是使用map容器实现的。
map<int, set<int>> idx;
在对m个货物项处理的过程中,每条货物清单都有一个bool数组存放该集装箱出现的货物的编号,然后对货物箱中的每个货物编号进行检查,如果该货物编号映射的集合非空,则检查其不相容的货物是否出现在集装箱。
这样算法的时间复杂度为O(MKlogN)
代码实现
#include <cstdio>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#include <algorithm>
#include <iostream>
#define ll long long
#define ep 1e-5
#define INF 0x7FFFFFFF
const int maxn = 100005;
const int maxm = 1005;
using namespace std;
int main() {
// 输入不相容对数和清单数
int n, m;
cin >> n >> m;
// 每个有不相容的货物编号都映射一个set容器,内部存放所有与之不相容的货物的编号
map<int, set<int>> idx;
while (n --) {
int a, b;
cin >> a >> b;
idx[a].insert(b);
idx[b].insert(a);
}
// 每条货物清单都有一个bool数组存放该集装箱出现的货物的编号
bool in_box[maxn];
while (m --) {
memset(in_box, 0, sizeof(in_box));
int num, staff[maxm];
cin >> num;
// 记录货物编号
for (int i = 0; i < num; i ++) {
cin >> staff[i];
in_box[staff[i]] = true;
}
// 对每个货物编号进行检查,如果该货物编号映射的集合非空,则检查其不相容的货物是否出现在集装箱
bool safe = true;
for (int i = 0; i < num; i ++) {
if (!idx[staff[i]].empty()) {
set<int>:: iterator it;
// 检查不相容货物
for (it = idx[staff[i]].begin(); it != idx[staff[i]].end(); it ++) {
if (in_box[*it]) {
safe = false;
break;
}
}
if (!safe) {
break;
}
}
}
// 输出
if (safe) {
cout << "Yes" << endl;
}
else {
cout << "No" << endl;
}
}
return 0;
}
总结
学习不息,继续加油