PAT A1149 Dangerous Goods Packaging (25 分) 模拟

    题目大意：给出N对不能相容的物品id，之后给出M个序列，判断这些序列中是否存在不相容的物品。

    想出来两种思路，测试了下第二种思路整体上快一些：

① 因为id是五位数，用set<int> incompatiable[MAXN]来存储每个货物的不相容id，之后对读入的序列，用二重循环遍历(i , j)，判断 j是否在 i 的不相容列表里，如果是则存在不相容物品。这种方法的最坏时间复杂度是O(N^2)量级的，不是太好。

AC代码：

#include <vector>
#include <cstdio>
#include <set>
using namespace std;

const int MAXN = 100010;
set<int> incompatible[MAXN];

int main()
{
    int N, M;
    scanf("%d%d", &N, &M);
    for (int i = 0; i < N; ++i)
    {
        int u, v;
        scanf("%d%d", &u, &v);
        incompatible[u].insert(v);
        incompatible[v].insert(u);
    }
    for (int query = 0; query < M; ++query)
    {
        int cnt;
        scanf("%d", &cnt);
        vector<int> v(cnt);
        for (int j = 0; j < cnt; ++j)
        {
            scanf("%d", &v[j]);
        }
        bool flag = true;
        for (int i = 0; i < v.size() - 1 && flag; ++i)
        {
            for (int j = i + 1; j < v.size() && flag; ++j)
            {
                if(incompatible[v[i]].count(v[j]) == 1)
                    flag = false;
            }
        }
        if(flag) printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}

② 用vector<int> G[MAXN]来存储不相容物品的列表，bool inq[MAXN]记录每个id是否在当前序列里。之后对每个在序列里的id，遍历它的不相容物品列表，如果也在当前序列里，则存在不相容的物品。

AC代码：

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdio>
using namespace std;

const int MAXN = 100010;
vector<int> G[MAXN];
bool inq[MAXN] = {false};
int main()
{
    int N, M;
    cin >> N >> M;
    for (int i = 0; i < N; ++i)
    {
        int u, v;
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    for (int q = 0; q < M; ++q)
    {
        fill(inq, inq + MAXN, 0);
        bool flag = true;
        int K;
        scanf("%d", &K);
        vector<int> vq;
        for (int i = 0; i < K; ++i)
        {
            int tmp;
            scanf("%d", &tmp);
            vq.push_back(tmp);
            inq[tmp] = true;
        }
        for (int i = 0; i < K && flag; ++i)
        {
            int now = vq[i];
            for (int j = 0; j < G[now].size() && flag; ++j)
            {
                if(inq[G[now][j]])
                    flag = false;
            }
        }
        if(flag) printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}