PAT (Basic Level) 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

分析：刚开始用数据结构暴力两个测试点没有通过（超时）。然后仔细想了想用一个set容器把不能在一起的储存起来，这样就方便查找，减少时间复杂度。 

代码：

#include<iostream>
#include<vector>
#include<set>
#include<string>
using namespace std;
int is_safe(vector<set<int>> v, int a[], int n)
{
	for (int i = 0; i < n-1; i++)
	{
		for (int j = i + 1; j < n; j++)
		{
			if (v[a[i]].find(a[j]) != v[a[i]].end())
				return 0;
		}
	}
	return 1;
}
int main()
{
	int n, m;
	cin >> n >> m;
	vector<set<int>> v(100005);
	int a, b;
	while (n--)
	{
		scanf("%d%d", &a, &b);
		v[a].insert(b);
		v[b].insert(a);
	}
	while (m--)
	{
		int x;
		cin >> x;
		int *a1 = new int[x];
		for (int i = 0; i < x; i++)
			scanf("%d", &a1[i]);
		//printf("%s\n", is_safe(v, a1, x) ? "Yes" : "No");
		if (is_safe(v, a1, x))
		{
			puts("Yes");
		}
		else
			puts("No");
	}
		return 0;
}