pat.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

 

开始的解法，写着写着就发现时间复杂度hold不住了妥妥的爆了，不过还是写完交了下果然4个样例2个超时。后来换了种方法。

开始的解法：

#include<iostream>
#include<map>
#include<vector>
#include<utility>
using namespace std;

int main()
{
	map<pair<int,int>,bool> m1;
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		pair<int,int> pa(a,b);
		m1[pa]=true;
	}
	for(int i=0;i<m;i++)
	{
		int k;
		cin>>k;
		vector<int> v;
		for(int j=0;j<k;j++)
		{
			int x;
			cin>>x;
			v.push_back(x);
		}
		int flag=1;
		for(int j=0;j<k;j++)
		{
			for(int p=j+1;p<k;p++)
			{
				pair<int,int> pa(v[j],v[p]);
				pair<int,int> pb(v[p],v[j]);
				if(m1[pa]==true||m1[pb]==true)
				{
					flag=0;
					break;
				}
			}
		}
		if(flag) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	
	return 0;
} 

 

正解： 

用一个set数组表示禁止的关系，这样查询某个物品所禁止的哪些物品时少了一个logn，如果最大数据的话log(5000x5000/2)，就这部分快了进十倍。。。

#include<iostream>
#include<set>
#include<vector>
using namespace std;

set<int> banset[100010];

int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=0;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		banset[a].insert(b);
		banset[b].insert(a);
	}
	for(int i=0;i<m;i++)
	{
		int k;
		cin>>k;
		vector<int> v;
		for(int j=0;j<k;j++)
		{
			int x;
			cin>>x;
			v.push_back(x);
		}
		bool flag=true;
		for(int j=0;j<k;j++)
		{
			for(int p=j+1;p<k;p++)
			{
				if(banset[v[j]].find(v[p])!=banset[v[j]].end())
				{
					flag=false;
					break;
				}
			}
			if(!flag) break;
		}
		if(flag) cout<<"Yes\n";
		else cout<<"No\n";
	}
	return 0;
}