PAT 乙级 1090.危险品装箱 C++/Java

题目来源

集装箱运输货物时，咱们必须特别当心，不能把不相容的货物装在一只箱子里。好比氧化剂绝对不能跟易燃液体同箱，不然很容易形成爆炸。

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

输入格式：

输入第一行给出两个正整数：N (≤104) 是成对的不相容物品的对数；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

分析：

输入N，M，分别为：

• N：不相容物品的对数
• M：物品清单单数

接下来N行，每行输入两个数，表示不相容的物品

接下来M行，每行给出一箱货物的清单

思路：

一开始用了unordered_map<int, unordered_map<int, int>> m来存放，超内存了

后来考虑用一个map和vector数组来存放不相容的物品

好比下面这个输入，咱们能够用unordered_map<int, vector<int>> m来存放

把第一第二个物品都放入map中，记录不相容的物品

20001 20002
20003 20004
20005 20006
20003 20001
20005 20004
20004 20006

这样就知道了每一个物品和哪些东西不相容：

key(int) -> value(vector<int>)

m[20001] -> 20002, 20003
m[20002] -> 20001
m[20003] -> 20001, 20004
m[20004] -> 20003, 20005, 20006
m[20005] -> 20004, 20006
m[20006] -> 20004, 20005

对于接下来的M行输入，每行的输入存入一个数组res

4 00001 20004 00002 20003
5 98823 20002 20003 20006 10010
3 12345 67890 23333

好比第一行的4 00001 20004 00002 20003

存入数组res获得：00001 20004 00002 20003

同时用一个数组arr记录这4个id的出现状况

好比（用下标记录）：

arr[00001] = 1
arr[20004] = 1
arr[00002] = 1
arr[20003] = 1

遍历数组res，在unordered_map中，获取到与res[i]不相容的物品

m[res[i]]就是不相容的物品，这是一个vector数组

• 好比res[i] = 20001：
  • m[res[i]] = m[20001] -> 20002, 20003

遍历这个数组，取出里面的元素x

• 若是arr[x] == 1：就说明有不相容的物品

C++实现：

#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_map>
#include <unordered_set>
#include <string>
#include <stack>
#include <cmath>
#include <map>

using namespace std;


int main() {
    short N;  // 成对的，不相容物品的对数
    short M;  // 集装箱货品清单的单数
    unordered_map<int, vector<int>>m;

    cin >> N >> M;

    for (int i = 0; i < N; ++i) {
        int first, second;
        cin >> first >> second;
        m[first].push_back(second);
        m[second].push_back(first);
    }

    while (M--) {
        bool flag = false;
        int cnt;
        cin >> cnt;
        vector<int> res(cnt);
        int arr[100001] = { 0 };
        for (int i = 0; i < cnt; ++i) {
            cin >> res[i];
            arr[res[i]] = 1;
        }
        for (int i = 0; i < cnt; ++i) {
            for (int j = 0; j < m[res[i]].size(); ++j) {
                // m[res[i]]是一个vector数组
                // m[res[i]][j]是vecotr里的元素
                int x = m[res[i]][j];
                if (arr[x] == 1) {
                    flag = true;
                }
            }
        }
        if (flag) {
            printf("No\n");
        }
        else {
            printf("Yes\n");
        }
    }

    return 0;
}

Java实现：

// TODO