PAT乙级 | 1090 危险品装箱 (25分)（优化思路详解）

集装箱运输货物时，我们必须特别小心，不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱，否则很容易造成爆炸。

本题给定一张不相容物品的清单，需要你检查每一张集装箱货品清单，判断它们是否能装在同一只箱子里。

输入格式：

输入第一行给出两个正整数：N (≤10⁴) 是成对的不相容物品的对数；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

思路：这个题有意思，博主一开始定义了一个二维的vector数组：v[100000]，后面显示运行错误，我猜想是可能分配的空间太大了，所以我们用map来模拟二维数组：map<int,vector > mp;，这不就有了吗，把互不相容的危险品都放一起，最后再判断即可，这里还要剪枝，否则是运行超时，在输入的过程中可以发现有的不是危险品，对于该类物品没必要进行判断，还有一旦找到危险品直接退出所有循环进行打印。

#include <iostream> #include <vector> #include <map> using namespace std; int main() { int n,m,k,a,b; bool hash[100000]={0}; map<int,vector<int> > mp; cin >>n>>m; for(int i=0;i<n;i++){ scanf("%d%d",&a,&b); hash[a]=hash[b]=true; //a和b是危险品 mp[a].push_back(b); //归类 mp[b].push_back(a); //归类 } for(int i=0;i<m;i++){ cin >>k; int flag=0; vector<int> v(k); for(int j=0;j<k;j++) scanf("%d",&v[j]); for(int j=0;j<k;j++){ //判断过程 if(hash[v[j]]){ for(int l=j+1;l<k;l++){ if(hash[v[l]]){ for(int q=0;q<mp[v[j]].size();q++){ if(mp[v[j]][q]==v[l]){ flag=1; break; } } if(flag==1) break; //剪枝 } } if(flag==1) break; //剪枝 } } printf("%s\n",flag==1?"No":"Yes"); //打印 } return 0; }