1090 危险品装箱

分数 25

全屏浏览题目切换布局

作者 CHEN, Yue

单位 浙江大学

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

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

输入格式：

输入第一行给出两个正整数：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

分析：

建立 map映射是解题的关键，在每次输入完一组物品后进行排查。如果出现不相容的物品就直接退出循环。

代码：

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>

using namespace std;

int main() {
    int N, M, item1, item2;
    map<int, vector<int> > itemToItems;

    cin >> N >> M;

    // 输入危机物品
    for (int i = 0; i != N; i++) {
        cin >> item1 >> item2;
        // 建立 map 映射
        itemToItems[item1].push_back(item2);
        itemToItems[item2].push_back(item1);
    }


    while (M--) {
        int cnt, flag = false, ans[100000] = { 0 };
        cin >> cnt;
        vector<int> items(cnt, 0);
        for (int i = 0; i != cnt; i++) {
            cin >> items[i];
            // 标记该物品出现过
            ans[items[i]] = 1;
        }

        
        for (int i = 0; i != cnt && !flag; i++) {
            // 遍历不相容物品数组
            for (int j = 0; j < itemToItems[items[i]].size(); j++) {
                // 如果同一集转箱子出现不相容物品标记 flag 为 true
                if (ans[itemToItems[items[i]][j]] == 1) {
                    flag = true;
                    break;
                }
            }
        }
        // 
        cout << (flag ? "No" : "Yes") << endl;

    }
    return 0;
}