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
posted @ 2020-05-22 10:32  47的菠萝~  阅读(193)  评论(0编辑  收藏  举报