题目如下
集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸。
本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否能装在同一只箱子里。
输入格式:
输入第一行给出两个正整数: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
解题思路
对于这题,我自己的想法是把输入特别是实际运输的货物用string表示,利用getline函数接收一行的数据保存下来,然后对相克的物品进行循环查找,利用string的find方法进行搜索,如果查找到,即返回值!=-1的时候表示不存在相克的物品这样循环下去。但是事实告诉我,这样做又超时了……
先贴出我的源码:
改进之前的源代码
#include<iostream>
#include<vector>
using namespace std;
struct unsafe{
string g1;
string g2;
};
int main()
{
int N,M;
cin>>N>>M;
vector<unsafe> vec;
for(int i=0;i<N;i++)
{
unsafe u;
cin>>u.g1>>u.g2;
vec.push_back(u);
}
getchar();
vector<string> result;
for(int i=0;i<M;i++)
{
string str;
getline(cin,str);
// if(str[0]=='1'||str[0]=='0')
// {
// result.push_back("Yes");
// cout<<"Yes"<<endl;
// }
// else
// {
int flag=0;
for(int j=0;j<vec.size();j++)
{
if(str.find(vec[j].g1)!=-1&&str.find(vec[j].g2)!=-1)
{
result.push_back("No");
flag=1;
break;
}
}
if(flag==0)
{
result.push_back("Yes");
}
// }
}
for(int i=0;i<result.size();i++)
{
cout<<result[i]<<endl;
}
}
运行超时,最后一个测试点过不去,没办法,只能换一种方法。具体可以参考https://www.cnblogs.com/zlrrrr/p/9650186.html
改进后的代码
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
int N, M;
map<int, vector<int> > mp;
int vis[maxn];
int main() {
scanf("%d%d", &N, &M);
for(int i = 0; i < N; i ++) {
int x, y;
scanf("%d%d", &x, &y);
mp[x].push_back(y);
mp[y].push_back(x);
}
while(M --) {
int ans;
scanf("%d", &ans);
memset(vis, 0, sizeof(vis));
vector<int> v(ans);
for(int i = 0; i < ans; i ++) {
scanf("%d", &v[i]);
vis[v[i]] = 1;
}
bool flag = true;
for(int i = 0; i < v.size(); i ++) {
for(int j = 0; j < mp[v[i]].size(); j ++)
if(vis[mp[v[i]][j]]) flag = false;
}
if(flag) printf("Yes\n");
else printf("No\n");
}
return 0;
}