1090 危险品装箱 (25 分)
集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸。
本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否能装在同一只箱子里。
输入格式:
输入第一行给出两个正整数: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
开始的解法,写着写着就发现时间复杂度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;
}