1. 程式人生 > >1090 危險品裝箱——C++實現

1090 危險品裝箱——C++實現

題目

1090 危險品裝箱 (25 point(s))

集裝箱運輸貨物時,我們必須特別小心,不能把不相容的貨物裝在一隻箱子裡。比如氧化劑絕對不能跟易燃液體同箱,否則很容易造成爆炸。

本題給定一張不相容物品的清單,需要你檢查每一張集裝箱貨品清單,判斷它們是否能裝在同一只箱子裡。

輸入格式:

輸入第一行給出兩個正整數: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

演算法

見到對映關係很容易想到map,但是這裡map用一對一肯定不行,因為輸入裡面可能會一對多。即一個物品可能與多個物品不相容。因此不能建立map<int,int>型別,而是要建立map<int,vector<int>>型別。

在查詢的時候,需要將M個的每一行存起來,然後查詢每一個對應的map,看是否能找到,找得到則有不相容,反之相容。這裡map裡面的value實際上是一個vector。

程式碼

#include <iostream>
#include <map>
#include <vector>
using namespace std;
int main(){
	int N,M,t1,t2;	cin>>N>>M;
	map<int,vector<int>> idx;
	for(int i=0;i<N;i++){
		cin>>t1>>t2;
		idx[t1].push_back(t2);
		idx[t2].push_back(t1);
	}
	for(int i=0;i<M;i++){
		int cnt,flag=0,a[100000]={0};
		cin>>cnt;
		vector<int> v(cnt);
		for(int i=0;i<cnt;i++){
			cin>>v[i];
			a[v[i]]=1;    //有的物品都設定為1
		} 
		for(int i=0;i<cnt;i++)
			for(int j=0;j<idx[v[i]].size();j++) //查詢每一個map的vector型value
				if(a[idx[v[i]][j]]==1){    //if search,it means not match
					flag=1;
					break;
				} 
		printf("%s\n",flag?"No":"Yes");
	}
	return 0;
}