1. 程式人生 > 實用技巧 >1090 危險品裝箱 (25分)

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();
    }
}