PAT乙级真题 — 1090 危险品装箱(java)

集装箱运输货物时，我们必须特别小心，不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱，否则很容易造成爆炸。

本题给定一张不相容物品的清单，需要你检查每一张集装箱货品清单，判断它们是否能装在同一只箱子里。

输入格式：

输入第一行给出两个正整数：N (≤104) 是成对的不相容物品的对数；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

思路：因为一种物品可能和多种物品相斥，所以用HashMap<Integer,HashSet<Integer>>来存储各物品和其排斥物品，然后对每项货物进行判断，定义一个Set集合，从头对该箱物品进行判断，如果该物品有排斥物，就把排斥物放入这个Set集合，后续在对新来的物品判断，如果已经在集合里面，就说明该箱货物有危险品。

运行代码：

import java.io.*;
import java.io.IOException;
import java.util.*;
public class Main {static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));static StreamTokenizer st = new StreamTokenizer(br);static int ini() throws IOException {st.nextToken();return (int)st.nval;}public static void main(String[] args) throws IOException {int N = ini(), M = ini();int[] dif = new int[100000];HashMap<Integer, HashSet<Integer>> goods = new HashMap<>();for (int i = 0; i < N; i++) {int a = ini(), b = ini();if(goods.containsKey(a)) {goods.get(a).add(b);}else {HashSet<Integer> hs = new HashSet();hs.add(b);goods.put(a, hs);}if(goods.containsKey(b)) {goods.get(b).add(a);}else {HashSet<Integer> hs = new HashSet<>();hs.add(a);goods.put(b,hs);}}for (int i = 0; i < M; i++) {int n = ini(), flag = 1;HashSet<Integer> al = new HashSet<>();for (int j = 0; j < n; j++) {int t = ini();if(goods.containsKey(t)) {if(al.contains(t)) {flag = 0;}else {al.addAll(goods.get(t));}}}System.out.printf("%s\n",flag == 1 ? "Yes":"No");}}
}

运行结果：