PAT 乙级 1090 危险品装箱

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

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

输入格式：

输入第一行给出两个正整数：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结构，主要是一个int型变量对应了多个int类型的集合，所以便想到了set容器以及vector容器。建立一个一对多关系的便是使用map<int,set<int> >型结构，也可以用map<int, vector<int> >型结构，即通过int下标可以对应出不同的容器结构，也可以调用出不同的函数。比如有一个物品为 j，要将物品a,b加入物品 j 的违禁品集合之中。需要的语句是 建立 map<int, set<int> >m，m[ j ].insert(a) ,m[ j ].insert(b).若是vector<int>,则对应的是m[ j ].push_back(a)。

值得注意的是，使用for循环遍历 j 对应的容器vector的元素，语句如下：

    for(j=0;j<k;j++)
        {
            for(int l=j;l<k;l++)
            {
                for(int jj=0;jj<m[wupin[j]].size();jj++)
                {
                    if(wupin[l]==m[wupin[j]][jj])
                    {
                        flag=1;
                    }
                }
            }
        }

直接写成m[ j ][ l ]的类似于二维数组的形式，可以访问每一个元素。

解决了采用的容器的问题后，需要解决的是两个超时间的问题，首先利用set结构，将出现的违禁品加入set结构s中，如果在物品输出中存在 s.count(a)==0，便表示这个物品不是违禁品，便没有必要再去将这个物品加入比较的数组wupin中，这里解决了第一个超时。数组wupin用于对有可能的违禁品进行比较，通过两个for循环，第一个for循环用于控制外层的物品，下标由 j 到 q。内层有j+1开始到k为止,一旦出现了违禁品，即m[wupin[j]].count(wupin[l])==1，即物品 l 是 物品 j 对应的违禁品。则控制flag为1，退出循环，在第二个层循环继续退出。这样便不会超时。

代码如下：

#include <bits/stdc++.h>

using namespace std;

map<int,set<int> >m;
set<int>s;

int main()
{
    int N,M;
    cin>>N>>M;
    int i,k,j,id;
    int a,b;
    for(i=0; i<N; i++)
    {
        cin>>a>>b;
        m[a].insert(b);
        m[b].insert(a);
        s.insert(a);
        s.insert(b);
    }
    int flag=0;
    int wupin[1010];
    int q;
    for(i=0; i<M; i++)
    {
        flag=0;
        q=0;
        cin>>k;
        for(j=0; j<k; j++)
        {
            cin>>id;
            if(s.count(id)==1)
            {
                wupin[q++]=id;//第一次省时，非违禁品不需要加入集合
            }
        }
        for(j=0; j<q; j++)
        {
            for(int l=j; l<q; l++)
            {
                if(m[wupin[j]].count(wupin[l])==1)
                {
                    cout<<"No"<<endl;
                    flag=1;
                    break;//第二次省时，出现冲突的物品直接退出
                }
            }
            if(flag==1)
            {
                break;
            }
        }
        if(flag==0)
        {
            cout<<"Yes"<<endl;
        }

    }
    return 0;
}

 下面是第一次用vector的代码，超时了两个测试点，为了不忘记 map<int,vector<int> >而记录

#include <bits/stdc++.h>

using namespace std;

map<int,vector<int> >m;

int main()
{
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(0);
    int N,M;
    cin>>N>>M;
    int i,k,j,id;
    int a,b;
    for(i=0;i<N;i++)
    {
        cin>>a>>b;
        m[a].push_back(b);
        m[b].push_back(a);
    }
    int flag=0;
    int wupin[1010];
    for(i=0;i<M;i++)
    {
        flag=0;
        cin>>k;
        for(j=0;j<k;j++)
        {
            cin>>id;
            wupin[j]=id;
        }
        for(j=0;j<k;j++)
        {
            for(int l=j;l<k;l++)
            {
                for(int jj=0;jj<m[wupin[j]].size();jj++)
                {
                    if(wupin[l]==m[wupin[j]][jj])
                    {
                        flag=1;
                    }
                }
            }
        }
        if(flag==0)
        {
            cout<<"Yes"<<endl;
        }
        else if(flag==1)
        {
            cout<<"No"<<endl;
        }
    }
    return 0;
}