题目链接:
描述
哈利波特在魔法学校的必修课之一就是学习魔咒。据说魔法世界有100000种不同的魔咒,哈利很难全部记住,但是为了对抗强敌,他必须在危急时刻能够调用任何一个需要的魔咒,所以他需要你的帮助。 给你一部魔咒词典。当哈利听到一个魔咒时,你的程序必须告诉他那个魔咒的功能;当哈利需要某个功能但不知道该用什么魔咒时,你的程序要替他找到相应的魔咒。如果他要的魔咒不在词典中,就输出“what?”
输入描述:
首先列出词典中不超过100000条不同的魔咒词条,每条格式为: [魔咒] 对应功能 其中“魔咒”和“对应功能”分别为长度不超过20和80的字符串,字符串中保证不包含字符“[”和“]”,且“]”和后面的字符串之间有且仅有一个空格。词典最后一行以“@END@”结束,这一行不属于词典中的词条。 词典之后的一行包含正整数N(<=1000),随后是N个测试用例。每个测试用例占一行,或者给出“[魔咒]”,或者给出“对应功能”。
输出描述:
每个测试用例的输出占一行,输出魔咒对应的功能,或者功能对应的魔咒。如果魔咒不在词典中,就输出“what?”
示例1
输入:
[expelliarmus] the disarming charm
[rictusempra] send a jet of silver light to hit the enemy
[tarantallegra] control the movement of one's legs
[serpensortia] shoot a snake out of the end of one's wand
[lumos] light the wand
[obliviate] the memory charm
[expecto patronum] send a Patronus to the dementors
[accio] the summoning charm
@END@
4
[lumos]
the summoning charm
[arha]
take me to the sky
输出:
light the wand
accio
what?
what?
思路:
- 创建一个
map
数据结构,用于存储魔咒与对应功能的映射关系。 - 通过循环读取输入,将词典中的魔咒和对应功能存入
map
中。 - 读取正整数
n
,表示接下来有n
个测试用例。 - 循环读取每个测试用例,对于每个测试用例
ss
:- 如果
ss
不在词典中,输出 "what?"。 - 如果
ss
在词典中且对应的映射值以 "[" 开头,说明ss
是功能,不是魔咒,输出对应功能,去掉中括号。 - 否则,输出魔咒对应的功能。
- 如果
- 循环结束,程序结束。
源代码:
#include <iostream>
#include <string>
#include <map>
using namespace std;
int main() {
string s;
map<string, string> maps; // 创建一个map,用于存储魔咒与对应功能的映射关系
while (getline(cin, s)) {
if (s == "@END@") { // 输入以"@END@"结束,退出循环
break;
}
int pos = s.find(']');
string s1 = s.substr(0, pos + 1); // 提取魔咒部分,包括中括号
string s2 = s.substr(pos + 2); // 提取功能部分,去除空格
maps[s1] = s2; // 将魔咒与功能的映射关系存入map
maps[s2] = s1; // 将功能与魔咒的映射关系存入map
}
int n;
cin >> n;
cin.ignore(); // 忽略掉之前输入时的换行符
for (int i = 0; i < n; i++) {
string ss;
getline(cin, ss); // 读取一个测试用例
if (maps.find(ss) == maps.end()) { // 如果魔咒不在词典中
cout << "what?" << endl; // 输出"what?"
}
else if (maps[ss][0] == '[') { // 如果找到的映射值以'['开头,说明ss是功能,不是魔咒
cout << maps[ss].substr(1, maps[ss].size() - 2) << endl; // 输出功能,去掉中括号
}
else {
cout << maps[ss] << endl; // 输出魔咒对应的功能
}
}
return 0;
}
提交结果: