魔咒词典

题目描述

       哈利波特在魔法学校的必修课之一就是学习魔咒。据说魔法世界有100000种不同的魔咒，哈利很难全部记住，但是为了对抗强敌，他必须在危急时刻能够调用任何一个需要的魔咒，所以他需要你的帮助。

      给你一部魔咒词典。当哈利听到一个魔咒时，你的程序必须告诉他那个魔咒的功能；当哈利需要某个功能但不知道该用什么魔咒时，你的程序要替他找到相应的魔咒。如果他要的魔咒不在词典中，就输出“what?”

输入

首先列出词典中不超过100000条不同的魔咒词条，每条格式为：

    [魔咒] 对应功能

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

输出

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

样例输入

[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?

#include <cstdio>
#include <cstring>
#include <string>
#include <vector>
using namespace std;
typedef struct {
	string name, func;
} dict_def;


vector<int> dict_indexed_by_name[26], dict_indexed_by_func[26];
vector<int>::iterator it_vec;
vector<dict_def> dict;
int  item_num;


void Parse(char p_input[]) {
	char s_name[20], s_func[80], *pos;
	pos = strchr(p_input,']');
	strcpy(s_func, pos + 2);
	*pos = '\0';
	strcpy(s_name, p_input + 1);
	string s1(s_name), s2(s_func);
	dict_def a_word;
	a_word.name = s1, a_word.func = s2;
	dict.push_back(a_word);
	dict_indexed_by_name[s_name[0] - 'a'].push_back(item_num);
	dict_indexed_by_func[s_func[0] - 'a'].push_back(item_num);
	item_num++;
}


string FindName(string p_func)
{
	int index = p_func[0] -'a';
	for (it_vec = dict_indexed_by_func[index].begin();
			it_vec < dict_indexed_by_func[index].end(); it_vec++)
		if (dict[(*it_vec)].func == p_func)
			break;
	if (it_vec < dict_indexed_by_func[index].end())
		return dict[(*it_vec)].name;
	return "";
}
string FindFunc(string p_name)
{
	int index = p_name[0] - 'a';
	for (it_vec = dict_indexed_by_name[index].begin();
			it_vec < dict_indexed_by_name[index].end(); it_vec++)
		if (dict[(*it_vec)].name == p_name)
			break;
	if (it_vec < dict_indexed_by_name[index].end())
		return dict[(*it_vec)].func;
	return "";
}
int main() {
	char input[100];
	while (gets(input) != NULL) {
		while (strcmp(input, "") == 0)
			gets(input);
		if (strcmp(input, "@END@") == 0)
			break;
		Parse(input);
	}
	int case_num;
	string s_name, s_func, ans;
	scanf("%d", &case_num);
	getchar();
	while (case_num--) {
		gets(input);
		if (input[0] == '[') {
			s_name = input;
			s_name.erase(s_name.begin());
			s_name.erase(s_name.end() - 1);
			ans = FindFunc(s_name);
			printf("%s\n", ans == "" ? "what?" : ans.c_str());
		} else {
			s_func = input;
			ans = FindName(s_func);
			printf("%s\n", ans == "" ? "what?" : ans.c_str());
		}
	}
	return 0;
}