PTA 1090 危险品装箱 (25 分) 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
结尾无空行

解题思路：

①简单暴力思路（测试点3过不去，显示超时）：
先用一个结构体数组来存危险品对，再利用货物来遍历结构体数组中的数，如果发现存在，则再利用货物中的其他货物来遍历是否等于结构体数组中的另一个数，如果相等，则说明含危险品

代码示例①：

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

using namespace std;

struct st //存危险品对
{
	string s1;
	string s2;
};

int main()
{
	int m, n;
	cin >> m >> n;

	vector<st> res; //存结构体

	while (m--)
	{
		st temp;
		cin >> temp.s1 >> temp.s2;

		res.push_back(temp);
	}

	while (n--)
	{
		int cnt;
		cin >> cnt;

		vector<string> ans; //存货物

		while (cnt--)
		{
			string s;
			cin >> s;
			ans.push_back(s);
		}

		int key = 1; //是否有危险品标志 为0表示有
		for (int i = 0; i < ans.size(); i++) //遍历货物
		{
			for (int j = 0; j < res.size(); j++)
			{
				//如果ans中的货物等于结构体数组中的某一个数
				//则就从ans中的下一个货物开始遍历看是否等于结构体数组中的另一个数
				if (ans[i] == res[j].s1 || ans[i] == res[j].s2)
				{
					for (int k = i + 1; k < ans.size(); k++)
					{
						if (ans[k] == res[j].s1 || ans[k] == res[j].s2)
						{
							key = 0;
							break;
						}
					}
				}
				if (key == 0)
				{
					break;
				}
			}
			if (key == 0)
			{
				break;
			}
		}

		if (key == 1)
		{
			cout << "Yes" << endl;
		}
		else
		{
			cout << "No" << endl;
		}
	}
}

运行结果：

②高级映射思路（完美通过）：
利用map容器建立一对多的关系来确立危险品之间的联系，再找货物中是否含有危险品即可
Tips：说起来简单，理解有点困难，仔细看代码

代码示例②：

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

using namespace std;

int main()
{
	int n, m;
	cin >> n >> m;

	map<int, vector<int>> res; //一个数对应一个数组 建立起一对多的关系
	while (n--)
	{
		int num1, num2;
		cin >> num1 >> num2;

		res[num1].push_back(num2);
		res[num2].push_back(num1);
	}
	/*
	此时res中的int：
	20001 对应数组： [20002, 20003]
	20002 对应数组： [20001]
	20003 对应数组： [20004, 20001]
	20004 对应数组： [20003, 20005, 20006]
	以此类推……
	*/

	while (m--)
	{
		int cnt;
		cin >> cnt;

		vector<int> temp;
		vector<int> ans(100005);
		while (cnt--)
		{
			int num;
			cin >> num;
			temp.push_back(num);
			ans[num] = 1;
		}

		/*
		此时：
		temp(ans的下标): 00001 20004 00002 20003
		ans:			   1     1     1     1
		*/

		int key = 1;
		for (int i = 0; i < temp.size(); i++)
		{
			for (int j = 0; j < res[temp[i]].size(); j++)
			{
				//在res中如果确定了有temp中的某一个数
				//此时这个数所对应的数组中的数如果也存在于temp中
				//那么在数组ans中就会找到他，就说明temp中有危险品
				if (ans[res[temp[i]][j]] == 1)
				{
					key = 0;
					break;
				}
			}
			if (key == 0)
			{
				break;
			}
		}

		if (key == 1)
		{
			cout << "Yes" << endl;
		}
		else
		{
			cout << "No" << endl;
		}
	}
}

运行结果：