PAT(Basic Level) 乙级 1090 危险品装箱

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

思路：

1.一个物体可能和多个物体不相容，故采用数组来存储多个与其不相容的物体。
2.为了方便可利用物体编号作为数组下标，找是否有与其不相容的物体时，便可通过该下标来找。
3.数组可采用C++ STL中vector：”变长数组”，因为要使用find()函数来找货物清单中的物体有没有与其不相容的，故使用set。
4.可能会遇到段错误，注意到每件物品用一个 5 位数的编号代表，故定义vector数组时设置为6位数。
PS: set可以自动去重且排序。

代码：

#include<cstdio>
#include<set>
#include<vector>
using namespace std;
bool if_warning(int h[],int k,vector<set<int> > vi){
    for(int i=0;i<k;i++){
        for(int j=i+1;j<k;j++){
            if(i != j && vi[h[i]].find(h[j]) != vi[h[i]].end()){
                return false;
            }
        }
    }
    return true;
}
int main(){
    vector<set<int> > vi(100010);
    int n,m,k;
    int a,b;
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++){
        scanf("%d %d",&a,&b);
        vi[a].insert(b);
        vi[b].insert(a);
    }
    int h[1010];
    for(int i=0;i<m;i++){
        scanf("%d",&k);
        for(int j=0;j<k;j++){
            scanf("%d",&h[j]);
        }
        if(if_warning(h,k,vi)==true){
            printf("Yes\n");
        }
        else printf("No\n");
    }



    return 0;


}