【JAVA】 PAT 乙级 1090 危险品装箱（踩点通过）

题目链接

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

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

输入格式：
输入第一行给出两个正整数：N (≤10⁴​​ ) 是成对的不相容物品的对数；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的value是一个set集合

Map<Integer, Set<Integer>> m = new HashMap<Integer, Set<Integer>>();

遍历的时候两个都要加到对方的危险集合里去

for (int i = 0; i < N; i++) {
	in.nextToken();
	int a = (int) in.nval;
	in.nextToken();
	int b = (int) in.nval;
	if (m.containsKey(a)) {
		m.get(a).add(b);
	} else {
		Set<Integer> ts = new HashSet<Integer>();
		ts.add(b);
		m.put(a, ts);
	}
	if (m.containsKey(b)) {
		m.get(b).add(a);
	} else {
		Set<Integer> ts = new HashSet<Integer>();
		ts.add(a);
		m.put(b, ts);
	}
}

检查的思想和【JAVA】PAT 乙级 1065 单身狗 类似，用一个w集合存会与船上的物品产生危险的物品。

for (int i = 0; i < M; i++) {
	in.nextToken();
	int K = (int) in.nval;
	Set<Integer> w = new HashSet<Integer>();
	boolean isSafe = true;
	for (int j = 0; j < K; j++) {
		in.nextToken();
		int g = (int) in.nval;
		if (m.containsKey(g)) {
			if (w.contains(g)) {
				isSafe = false;
			} else {
				w.addAll(m.get(g));
			}
		}
	}
	out.println(isSafe ? "Yes" : "No");
}

完整AC代码

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.StreamTokenizer;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.HashSet;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
		StreamTokenizer in = new StreamTokenizer(bf);
		in.nextToken();
		int N = (int) in.nval;
		in.nextToken();
		int M = (int) in.nval;
		Map<Integer, Set<Integer>> m = new HashMap<Integer, Set<Integer>>();
		for (int i = 0; i < N; i++) {
			in.nextToken();
			int a = (int) in.nval;
			in.nextToken();
			int b = (int) in.nval;
			if (m.containsKey(a)) {
				m.get(a).add(b);
			} else {
				Set<Integer> ts = new HashSet<Integer>();
				ts.add(b);
				m.put(a, ts);
			}
			if (m.containsKey(b)) {
				m.get(b).add(a);
			} else {
				Set<Integer> ts = new HashSet<Integer>();
				ts.add(a);
				m.put(b, ts);
			}

		}
		for (int i = 0; i < M; i++) {
			in.nextToken();
			int K = (int) in.nval;
			Set<Integer> w = new HashSet<Integer>();
			boolean isSafe = true;
			for (int j = 0; j < K; j++) {
				in.nextToken();
				int g = (int) in.nval;
				if (m.containsKey(g)) {
					if (w.contains(g)) {
						isSafe = false;
					} else {
						w.addAll(m.get(g));
					}
				}
			}
			out.println(isSafe ? "Yes" : "No");
		}
		out.flush();
	}
}