【JAVA】 PAT 乙级 1090 危险品装箱(踩点通过)
题目链接
集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸。
本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否能装在同一只箱子里。
输入格式:
输入第一行给出两个正整数: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
用一个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();
}
}