Week15 A ZJM 与霍格沃兹

ZJM 与霍格沃兹

ZJM 为了准备霍格沃兹的期末考试，决心背魔咒词典，一举拿下咒语翻译题
题库格式：[魔咒] 对应功能
背完题库后，ZJM 开始刷题，现共有 N 道题，每道题给出一个字符串，可能是 [魔咒]，也可能是对应功能
ZJM 需要识别这个题目给出的是 [魔咒] 还是对应功能，并写出转换的结果，如果在魔咒词典里找不到，输出 “what?”
Input

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

Output

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

Sample Input

[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

Sample Output

light the wand
accio
what?
what?

问题分析

题意非常好理解，利用map结构将功能和咒语对应起来即可。一开始是直接存储字符串map<string,string>，但是会MLE。
改成map<long long ,string>，将字符串转换为hash值存储。求hash值：
建立两个map数组存储字典，map1：key为咒语，value为功能；map2：key为功能，value为咒语。

输入要判断的字符串，先判断是咒语还是功能，在计算hash值，如果是咒语，在map1中寻找其功能；如果是功能，在map2中寻找其咒语。

代码实现

#include<iostream>
#include<map>
#include<string>
#include<math.h>
using namespace std;

map<long long, string> book1,book2; 
map<long long, string>::iterator it;
long long mod = 100000007 ;
int n;
string s,str;

int main()
{
	getline(cin,s);
	long long hash1,hash2;
	while(s!="@END@")
	{
		hash1=hash2=0;
		int in=s.find(']');
		string temp = s.substr(1,in-1);
		str = s.substr(in+2);
		long long seed=7;
		for(int i=0; i<temp.size(); i++)
		{
			if(temp[i]!=' ')
			{
				hash1+=((temp[i])*seed)%mod;
				seed = seed*7%mod;
			}
			    
		}
		seed=7;    
		for(int i=0; i<str.size(); i++)
		{
			if(str[i]!=' ')
			{
			    hash2+=((str[i])*seed)%mod;	
			    seed=seed*7%mod;
			}
			
		}
	//	cout<<"hash1:"<<hash1<<endl;
	//	cout<<"hash2:"<<hash2<<endl;
		book1.insert(pair<long long,string>(hash1,str));
		book2.insert(pair<long long,string>(hash2,temp));
	//	it=book1.find(hash1);
	//	cout<<it->second<<endl;
		getline(cin,s);
	}
	cin>>n;
	getchar();
	for(int i=0; i<n; i++)
	{
		string temp;
		getline(cin,temp);
		bool d=true;
		long long hash=0;
		if(temp[0]=='[')
		{
			
			int len=temp.size();
			string ss=temp.substr(1,len-2);
			long long seed=7;
			for(int i=0; i<ss.size(); i++)
			{
				if(ss[i]!=' ')
				{
					hash+=((ss[i])*seed)%mod;
					seed=seed*7%mod;
				}
				    
			}
		//    cout<<hash<<endl;
			it=book1.find(hash);
			if(it != book1.end())
			{
				cout<<it->second<<endl;
			}
			else
			    cout<<"what?"<<endl;
		}else{
			long long seed=7;
			for(int i=0; i<temp.size(); i++)
	    	{
		    	if(temp[i]!=' ')
		    	{
			    	hash+=((temp[i])*seed)%mod;
			    	seed = seed*7%mod;
		    	} 
	     	}
	    // 	cout<<hash<<endl;
		    it=book2.find(hash);
			if(it != book2.end())
			{
				cout<<it->second<<endl;
			}
			else
			    cout<<"what?"<<endl;
		}
		
	}
}