【PAT乙级】1090 危险品装箱

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

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

输入格式:

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

个人思路

这题在考试的时候我没做出来,今天写的时候对STL的各种工具有了进一步了解,所以本题主要是使用了map和set容器解决的。

每个有不相容的货物编号都映射一个set容器,内部存放所有与之不相容的货物的编号,而这个映射的关系是使用map容器实现的。

map<int, set<int>> idx;

在对m个货物项处理的过程中,每条货物清单都有一个bool数组存放该集装箱出现的货物的编号,然后对货物箱中的每个货物编号进行检查,如果该货物编号映射的集合非空,则检查其不相容的货物是否出现在集装箱。

这样算法的时间复杂度为O(MKlogN)

代码实现

#include <cstdio>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#include <algorithm>
#include <iostream>
#define ll long long
#define ep 1e-5
#define INF 0x7FFFFFFF

const int maxn = 100005;
const int maxm = 1005;

using namespace std;

int main() {
    // 输入不相容对数和清单数
    int n, m;
    cin >> n >> m;
    
    // 每个有不相容的货物编号都映射一个set容器,内部存放所有与之不相容的货物的编号
    map<int, set<int>> idx;
    while (n --) {
        int a, b;
        cin >> a >> b;
        idx[a].insert(b);
        idx[b].insert(a);
    }
    
    // 每条货物清单都有一个bool数组存放该集装箱出现的货物的编号
    bool in_box[maxn];
    while (m --) {
        memset(in_box, 0, sizeof(in_box));
        int num, staff[maxm];
        cin >> num;
        // 记录货物编号
        for (int i = 0; i < num; i ++) {
            cin >> staff[i];
            in_box[staff[i]] = true;
        }
        // 对每个货物编号进行检查,如果该货物编号映射的集合非空,则检查其不相容的货物是否出现在集装箱
        bool safe = true;
        for (int i = 0; i < num; i ++) {
            if (!idx[staff[i]].empty()) {
                set<int>:: iterator it;
                // 检查不相容货物
                for (it = idx[staff[i]].begin(); it != idx[staff[i]].end(); it ++) {
                    if (in_box[*it]) {
                        safe = false;
                        break;
                    }
                }
                if (!safe) {
                    break;
                }
            }
        }
        // 输出
        if (safe) {
            cout << "Yes" << endl;
        }
        else {
            cout << "No" << endl;
        }
    }
    return 0;
}

总结

学习不息,继续加油

猜你喜欢

转载自blog.csdn.net/qq_34586921/article/details/83478798