PAT (Basic Level) Practice 1090

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

分析：用map记录危险品的映射关系，然后判断是否符合标准。一开始是遍历所有货物，判断其是否有危险品被放置在一起，但这样处理时间复杂度很高，很容易超时。后来直接遍历危险品映射关系，判断其中是否出现过所要装的货物，这种做法在时间和空间复杂度上较前者均有大幅减少。

代码：

#pragma warning(disable:4996)
#include<iostream>
#include<algorithm>
#include<vector>
#include<unordered_map>
#include<string>
using namespace std;
int main() {
	int N, M;
	cin >> N >> M;
	unordered_map<string, vector<string>> m;
	unordered_map<string, bool> mb;
	for (int i = 0; i < N; i++) {
		string s1, s2;
		s1.resize(5);
		s2.resize(5);
		scanf("%s %s", &s1[0], &s2[0]);
		m[s1].push_back(s2);
		m[s2].push_back(s1);
	}
	for (int i = 0; i < M; i++) {
		int K;
		scanf("%d", &K);
		int flag = 0;
		vector<string> v(K);
		mb.clear();
		for (int j = 0; j < K; j++) {
			v[j].resize(5);
			scanf("%s", &v[j][0]);
			mb[v[j]] = true;
		}
		for (int j = 0; j < v.size(); j++) {
			for (int k = 0; k < m[v[j]].size(); k++) {
				if (mb[m[v[j]][k]] == true) {
					flag = 1;
					break;
				}
			}
			if (flag == 1) {
				break;
			}
		}
		if (flag == 1) {
			printf("No\n");
		} else {
			printf("Yes\n");
		}
	}
	return 0;
}