蓝桥杯 ALGO-94 新生舞会

  算法训练 新生舞会  

时间限制：1.0s   内存限制：512.0MB

    

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

输入
　　第一行一个整数n（2<=n<=1000），表示学生人数。接下来的n行每行依次包含一名新生的姓名、学号、性别，分别用一个空格隔开。
　　之后的一行是一个整数m(1<=m<=1000)，表示询问的数目。接着的m行每行包含两个信息（姓名或学号），保证两个信息不属于同一人，中间用一个空格隔开。

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


样例输入
　　4
　　John 10 M
　　Jack 11 M
　　Kate 20 F
　　Jim 21 M
　　3
　　John 11
　　20 Jack
　　Jim Jack


样例输出
　　N
　　Y
　　N

分析：把姓名学号当字符串存储，用map来存储映射关系。

代码：

#include<iostream>
#include<algorithm>
#include<map>
#include<string>
using namespace std;
int main() {
	int n, m;
	map<string, char> mp;
	cin >> n;
	for (int i = 0; i < n; i++) {
		string name;
		string num;
		char sex;
		cin >> name >> num >> sex;
		mp[name] = sex;
		mp[num] = sex;
	}
	cin >> m;
	for (int i = 0; i < m; i++) {
		int flag = 0;
		string s1;
		string s2;
		cin >> s1 >> s2;
		if (mp[s1] == mp[s2]) {
			flag = 0;
		} else {
			flag = 1;
		}
		printf("%c\n", flag == 1 ? 'Y' : 'N');
	}
	return 0;
}