Java实现 蓝桥杯VIP 算法训练 新生舞会

问题描述

　　新生舞会开始了。n名新生每人有三个属性：姓名、学号、性别。其中，姓名用长度不超过20的仅由大小写字母构成的字符串表示，学号用长度不超过10的仅由数字构成的字符串表示，性别用一个大写字符‘F’或‘M’表示。任意两人的姓名、学号均互不相同。换言之，每个人可被其姓名或学号唯一确定。给出m对两人的信息（姓名或学号），判断他们是否能共舞。两人能共舞的充要条件为两人性别相异。

输入

　　第一行一个整数n（2<=n<=1000），表示学生人数。接下来的n行每行依次包含一名新生的姓名、学号、性别，分别用一个空格隔开。

　　之后的一行是一个整数m(1<=m<=1000)，表示询问的数目。接着的m行每行包含两个信息（姓名或学号），保证两个信息不属于同一人，中间用一个空格隔开。

输出

　　对于每个询问输出一行，如果两人可以共舞，输出一个大写字母‘Y’，否则输出一个大写字母‘N’。

---

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class 新生舞会 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		Map<String,String> map = new HashMap<String,String>();
		for (int i = 0; i <n; i++) {
		String a = sc.next();
		String b = sc.next();
		String c=sc.next();
		map.put(a, c);
		map.put(b, c);
		}
		int m = sc.nextInt();
		for (int i = 0; i <m; i++) {
			String a = sc.next();
			String b = sc.next();
			String s1=map.get(a);
			String s2=map.get(b);
			if (!s1.equals(s2)) {
				System.out.println("Y");
			}
			else {
				System.out.println("N");
			}

		}
	}

}