PAT乙级---危险品装箱（Set红黑树）

PAT乙级—危险品装箱

做这题时看到好多都是用STL中的map和vector写的…我实在是理解不来那种写法QAQ，既然都是暴力，何不使用set更暴力点O0O
下面就给大家感受下set的"暴力解法"QAQ

背景

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

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

输入格式：

输入第一行给出两个正整数：N (≤10⁴)是成对的不相容物品的对数；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. 使用set，每个下标对应此货物，里面存放了不相容的一系列货物，以便后面比对
4. 每个都进行比对，直接两个循环超级暴力解题…

下面解释下这两个for循环

for(l=0;l<k;l++)
        {
            for(j=0;j<k;j++)//
            {
                if(l != j and s[a[l]].size()>0 and s[a[l]].find(a[j]) != s[a[l]].end())
                {
                    flag = 1;
                    break;
                }
            }
            if(flag == 1)
                break;
        }

首先是第一个for，对应了一个货物的id，然后使用此id对应set里面的不相容的货物
第二个for，用于在此 a[j] 货物里头查找有没有在 a[l] 不相容列表中里面

下面是AC代码

#include <iostream>
#include <set>
using namespace std;
int main()
{
    set<int> s[100000];
    int n,i,j,m;
    cin>>n>>m;
    for(i=0;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        s[a].insert(b);
        s[b].insert(a);
    }
    for(i=0;i<m;i++)
    {
        int k,flag=0;
        cin>>k;
        int a[k];
        for(j=0;j<k;j++)
        {
            cin>>a[j];
        }
        int l;
        for(l=0;l<k;l++)
        {
            for(j=0;j<k;j++)
            {
                if(l != j and s[a[l]].size()>0 and s[a[l]].find(a[j]) != s[a[l]].end())
                {
                    flag = 1;
                    break;
                }
            }
            if(flag == 1)
                break;
        }
        if(!flag)
            cout<<"Yes"<<endl;
        else
            cout<<"No"<<endl;
    }
    return 0;
}

------------------------------------------------------------------end------------------------