魔咒词典

3. K12196 魔咒词典

题目描述

哈利波特在魔法学校的必修课之一就是学习魔咒。据说魔法世界有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 accio the summoning charm @END@ 4 lumos the summoning charm arha take me to the sky

输出样例1：复制

light the wand accio what? what?

说明

测试点数据保证，没有任何一个“魔咒”和“魔咒功能”是相同的字符串。

【耗时限制】1000ms 【内存限制】128MB

---

解答

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
map<string,string> a,b;
string s,s1,s2;
LL n;
int main(){
	while(cin>>s1 && s1!="@END@"){
		getline(cin,s2);
		s2.erase(0,1);
		a[s1]=s2;
		b[s2]=s1;
	}
	cin>>n;
	cin.get();
	while(n--){
		getline(cin,s);
		if(a.find(s)!=a.end()) cout<<a[s]<<endl;
		else if(b.find(s)!=b.end())cout<<b[s]<<endl;
		else cout<<"what?\n";
	}
	return 0;
}