PAT乙级1090(C++)——龙哥哥的刷题路

1090 危险品装箱 (25分)
集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸。
本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否能装在同一只箱子里。

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

解题思路:用multimap来存取不相容物品的对应关系。当一行货物清单输入完时,遍历该货物清单,对每一个货物,从multimap中得到不相容的物品,然后遍历货物清单,看是否有不相容的物品,如果有输出No,如果没有输出Yes

#include
#include
#include
#include
using namespace std;
bool search(int x, vector<int> y){
     
	for (int i = 0; i < y.size(); i++){
     
		if (y[i] == x){
     
			return true;
		}
	}
	return false;
}
int main(){
     
	int N = 0, M = 0, a = 0, b = 0;
	multimap<int, int> notMap;
	cin >> N >> M;
	for (int i = 0; i < N; i++){
     
		cin >> a >> b;
		notMap.insert({
      a, b });
	}
	for (int i = 0; i < M; i++){
     
		bool flag = 0;
		int num = 0, c = 0;
		vector<int> things;
		cin >> num;
		for (int j = 0; j < num; j++){
     
			cin >> c;
			things.push_back(c);
		}
		for (int k = 0; k < num && !flag; k++){
     
			auto entries = notMap.count(things[k]);
			if (entries != 0){
     
				auto iter= notMap.find(things[k]);
				while (entries){
     
					int x = iter->second;
					if (search(x, things)){
     
						printf("No\n");
						flag = 1;
						break;
					}
					entries--;
				}
			}
			else{
     
				continue;
			}
		}
		if (!flag){
     
			printf("Yes\n");
		}
    }
	return 0;
}

你可能感兴趣的:(pat乙级,算法,c++)