PAT A1149 Dangerous Goods Packaging (25)

PTA跳转：原题链接

本题大意是：给出不能放在一起的物品的列表，判断箱子里的物品能否一起存放。（原题直译应该是集装箱，个人感觉箱子更好理解）

输入：先输入整型N和M，接下来输入N对不能放在一起的物品。然后，输入M行数据（每行的第一个数字num表示该箱子里有num件物品，后面跟着num个数据）。

输出：若箱子里的物品能放在一起，输出"Yes"，否则输出"No"。

本题考点是map映射，“map<int, vector<.int> >”（请自行去掉加粗字体中第二个int前面的点。由于是标识符，若去掉，在文章中无法显示）既实现一对多的映射关系，又节省大量空间。

代码如下：

#include <iostream>
#include <vector>
#include <map>
using namespace std;

int main()
{
    int N, M;
    map<int, vector<int> > m;	//一对多，因为物品a可能既不能和物品b在一起，也不能和物品c在一起 
    cin >> N >> M;
	int goods1, goods2;
    while(N--)	//输入N对不能放在一起的物品 
	{
        cin >> goods1 >> goods2;
        m[goods1].push_back(goods2);	//互相记录下不能和哪些物品放在一起 
        m[goods2].push_back(goods1);
    }
    while(M--)	//输入M个箱子存放的物品，判断是否能够放在一起 
	{
        int num, flag = 1, a[100000] = {0};	//flag=1表示物品可以放在一起 
        scanf("%d", &num);	//这个箱子有num件物品 
        vector<int> v(num);
        for(int i = 0; i < num; i++)
		{
            scanf("%d", &v[i]);	//输入该箱子每件物品并把它状态设为1 
            a[v[i]] = 1;
        }
        for(int i = 0; i < v.size(); i++)	//若该箱子上里货物数量为n，就循环n次 
        {
            for(int j = 0; j < m[v[i]].size(); j++)	//若该箱子里的第i+1个货物有m件不能放在一起的物品，就循环m次 
            {
				//v[i]表示该箱子里的第i+1件物品
				//m[v[i]]表示不能与该物品一起存放的物品列表
				//m[v[i]][j]示不能与该物品一起存放的第j+1件物品 
				//a[m[v[i]][j]]判断该物品是否存在 
                if(a[m[v[i]][j]] == 1)
					flag = 0;
			}
		}
		string s = flag ? "Yes" : "No";
        cout << s;
        if(M) cout << endl;	//最后一行不换行 
        v.clear();
    }
    return 0;
}

样例输入：
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

---

氷鸢鸢鸢
2020.5.5