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

解析

  由于一个物品存在多个不相容物品,用一个map存储,且值为一个vector,作为某物品的不相容物品列表
  用re保存某件集装箱的物品,用aa表示某件物品是否出现
  isBan()函数判断是否有不相容物品存在,对于某集装箱存在物品re[i],遍历其不相容物品列表s[re[i]],若列表中的某个元素s[re[i]][j]存在,即aa[s[re[i]][j]] == 1,则不能安全运输
  若遍历完不相容列表都没有存在的,则能安全运输,对于每个集装箱要再判断完之后清空aa

答案

#include<iostream>
#include<iostream>
#include<string>
#include<algorithm>
#include<map>
#include<cstdio>
#include<vector>
using namespace std;

int re[10001];
map <int , int> aa;
map <int , vector<int>> s;
int n,m;
bool isBan(){
    for(int i = 0 ; i < n; i ++){
        for(int j = 0 ; j < s[re[i]].size() ; j ++){
            if(aa[s[re[i]][j]] == 1) return false;
        }
    }
    return true;
}
int main(){
    cin >> n >> m;
    int a,b;
    while(n--){
        scanf("%d%d",&a,&b);
        s[a].push_back(b);
        s[b].push_back(a);
    }
    while(m--){
        scanf("%d",&n);
        
        for(int i = 0 ; i < n ; i ++){
            scanf("%d",&re[i]);
            aa[re[i]] = 1;
        }

        if(isBan()){
            printf("Yes\n");
        }else{
            printf("No\n");
        }
        aa.clear();
    }
}
上一篇:PAT(Basic Level) Practice : 1090 危险品装箱 (25分)


下一篇:关于C++ scanf的一个小知识