【簡便解法】1090 危險品裝箱 (25分)_33行程式碼AC
阿新 • • 發佈:2021-01-23
立志用最少的程式碼做最高效的表達
PAT乙級最優題解——>傳送門
集裝箱運輸貨物時,我們必須特別小心,不能把不相容的貨物裝在一隻箱子裡。比如氧化劑絕對不能跟易燃液體同箱,否則很容易造成爆炸。
本題給定一張不相容物品的清單,需要你檢查每一張集裝箱貨品清單,判斷它們是否能裝在同一只箱子裡。
輸入格式:
輸入第一行給出兩個正整數: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
分析
不同於之前的一對一對映, 本題考查一對多對映。 其餘不變。
程式碼
#include<bits/stdc++.h>
using namespace std;
int main() {
ios::sync_with_stdio(false);
int n, m; cin >> n >> m;
unordered_map<int, unordered_set<int> >um;
for(int i = 0; i < n; i++) {
int x1, x2; cin >> x1 >> x2;
um[x1].insert(x2);
um[x2].insert(x1);
}
while(m--){
int k, a; cin >> k;
unordered_set<int>us;
for(int i = 0; i < k; i++) {
cin >> a;
us.insert(a);
}
for(auto i : us) { //遍歷
if(um.count(i)) { //如果有不相容
for(auto j : um[i]) { //遍歷不相容的物品
if(us.count(j)) { //不相容的物品在該箱中
cout << "No\n";
goto loop; //回到loop標記,適用於多重迴圈
}
}
}
}
cout << "Yes\n";
loop:;
}
return 0;
}