week15作業A ZJM 與霍格沃茲

ZJM 爲了準備霍格沃茲的期末考試，決心背魔咒詞典，一舉拿下咒語翻譯題

題庫格式：[魔咒] 對應功能

背完題庫後，ZJM 開始刷題，現共有 N 道題，每道題給出一個字符串，可能是 [魔咒]，也可能是對應功能

ZJM 需要識別這個題目給出的是 [魔咒] 還是對應功能，並寫出轉換的結果，如果在魔咒詞典裏找不到，輸出 “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?

建立兩個map，m1和m2，通過Bkdr Hash算法算出魔咒和對應功能的哈希值hash1和hash2
m1[hash1]=index，str1[index]=魔咒
m2[hash2]=index，str2[index]=對應功能
根據查詢的是魔咒還是對應功能在對應的map中查詢，如果map.find(hash)返回的結果是map.end（），則說明這個魔咒或功能在魔咒詞典中不存在，輸出what？；否則，返回的是數組中相應的索引index，輸出即可。
在計算哈希值時，因爲字符串最長爲80，所以可以用快速冪來所短時間；當然也可以不用，只不過從seed^1開始逐漸累加即可。

#include<iostream>
#include<map>
#include<string>
#include<string.h>
#define ll long long
#define seed 7
using namespace std;
int N,len;
ll mod=1e9+7;
string str1[110000],str2[110000];
map<ll,int> m1,m2;
ll quick_pow(int x,int a)
{//快速冪 
	ll ans=1;
	while(x>0)
	{
		if(x&1) ans=ans*a;
		a=a*a;
		x>>=1;
	}
	return ans;
}
ll func(string s)
{
	ll tmp=0;
	for(int i=s.size()-1;i>=0;i--)
	{
		int t=s.size()-i;
		tmp=tmp+quick_pow(t,seed)*(ll)s[i];
		tmp=tmp%mod;
	}
	tmp%=mod;
	return tmp;
}
int main()
{
	int cnt=0;
	while(1)
	{
		string str;
		getline(cin,str);
		if(str=="@END@")
			break;
		int flag=0;
		for(int i=1;i<str.size();i++)
		{
			if(flag==0&&str[i]!=']')
			{
				str1[cnt]+=str[i];
			}
			else if(flag==0&&str[i]==']')
			{
				i++;
				flag=1;
			}
			else if(flag==1)
			{
				str2[cnt]+=str[i];
			}
		}
		ll tmp1=func(str1[cnt]),tmp2=func(str2[cnt]);
		m1[tmp1]=cnt;
		m2[tmp2]=cnt; 
		cnt++;
	}
	cin>>N;
	string str;
	ll tmp;
	getchar();
	for(int i=0;i<N;i++)
	{
		getline(cin,str);
		if(str[0]=='[')
		{
			str.erase(str.size()-1,1);
			str.erase(0,1);
			tmp=func(str);
			if(m1.find(tmp)==m1.end())
			{
				cout<<"what?\n";
			}
			else
			{
				int index=m1[tmp];
				cout<<str2[index]<<'\n';
			}
		}
		else
		{
			tmp=func(str);
			if(m2.find(tmp)==m2.end())
			{
				cout<<"what?\n";
			}
			else
			{
				int index=m2[tmp];
				cout<<str1[index]<<'\n';
			}
		}
	}
}