魔咒词典（浙大复试上机）

前言：

21考研，记录一下准备复试路上写下的垃圾代码。本来啃《算法笔记》，但是感觉太多了做不完，改做王道机试指南。

题目描述：

哈利波特在魔法学校的必修课之一就是学习魔咒。据说魔法世界有100000种不同的魔咒，哈利很难全部记住，但是为了对抗强敌，他必须在危急时刻能够调用任何一个需要的魔咒，所以他需要你的帮助。 给你一部魔咒词典。当哈利听到一个魔咒时，你的程序必须告诉他那个魔咒的功能；当哈利需要某个功能但不知道该用什么魔咒时，你的程序要替他找到相应的魔咒。如果他要的魔咒不在词典中，就输出“what?”

输入描述

首先列出词典中不超过100000条不同的魔咒词条，每条格式为：
[魔咒] 对应功能
其中“魔咒”和“对应功能”分别为长度不超过20和80的字符串，字符串中保证不包含字符“[”和“]”，且“]”和后面的字符串之间有且仅有一个空格。词典最后一行以“@END@”结束，这一行不属于词典中的词条。
词典之后的一行包含正整数N（<=1000），随后是N个测试用例。每个测试用例占一行，或者给出“[魔咒]”，或者给出“对应功能”。

输出描述:

每个测试用例的输出占一行，输出魔咒对应的功能，或者功能对应的魔咒。如果魔咒不在词典中，就输出“what?”

解答

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<map>
#include<string>
using namespace std;
int main()
{
	string str;
	map<string, string>posi;
	map<string, string>reverse;
	int n;
	while (getline(cin,str)) {
		if (str == "@END@")
			break;
		else {
			string spell;
			string trans;
			spell = str.substr(0, str.find("]") + 1);
			str.erase(0, str.find("]") + 2);
			trans = str;
			posi.insert(pair<string, string>(spell, trans));
			reverse.insert(pair<string, string>(trans, spell));
		}
	}
	cin >> n;
	getchar();
	while (n--) {
		string key;
		getline(cin, key);
		if (key[0] == '[')
			if (posi[key] != "")
				cout << posi[key] << endl;
			else cout << "what?" << endl;
		else {
			string res = reverse[key];
			if (res != "") {
				res.erase(0, 1);
				res.erase(res.length() - 1, 1);
				cout << res << endl;
			}
			else
				cout << "what?" << endl;
		}
	}
}