PAT乙级-分类-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

原始方法：最后一个测试点超时，得22分

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    map<int,vector<int>>mp;
    int x,y;
    map<int,bool>flag;
    for(int i=0;i<n;i++)
    {
        cin>>x>>y;
        mp[x].push_back(y);
        mp[y].push_back(x);
        flag[x]=true;
        flag[y]=true;
    }
    int k;
    while(m--)
    {
        cin>>k;
        vector<int> v(k),f;
        int repel=0;
        for(int i=0;i<k;i++)
        {
            cin>>v[i];
            if(flag[v[i]]==true)
            {
                f.push_back(v[i]);
            }
        }

        for(int i=0;i<f.size();i++)
        {
            int j=0;
            for(j=0;j<f.size();j++)
            {
                if(j==i)
                    continue;
                vector<int>::iterator s=find(mp[f[i]].begin(),mp[f[i]].end(),f[j]);
                if(s==mp[f[i]].end())
                    continue;
                else
                {
                    repel=1;
                    break;
                }
            }

        }
        if(repel)
            printf("No\n");
        else
            printf("Yes\n");
    }
    return 0;
}

方法二：上网搜
改进：最后遍历时去搜每个可能相斥的物体的所有相斥物体，如果此物体也在该序列中则NO

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    map<int,vector<int>>mp;
    int x,y;
    map<int,bool>flag;
    for(int i=0;i<n;i++)
    {
        cin>>x>>y;
        mp[x].push_back(y);
        mp[y].push_back(x);
        flag[x]=true;
        flag[y]=true;
    }
    int k;
    while(m--)
    {
        cin>>k;
        vector<int> v(k),f;
        int repel=0;
        int a[100000]={0};
        for(int i=0;i<k;i++)
        {
            cin>>v[i];
            a[v[i]]=1;
            if(flag[v[i]]==true)
            {
                f.push_back(v[i]);//可能相斥序列
            }
        }

        for(int i=0;i<f.size();i++)
        {
            for(int j=0;j<mp[f[i]].size();j++)
            {
                if(a[mp[f[i]][j]]==1)
                    repel=1;
            }

        }
        if(repel)
            printf("No\n");
        else
            printf("Yes\n");
    }
    return 0;
}