程序设计思维与实践 week15

作业

A - ZJM 与霍格沃兹

题目：

https://vjudge.net/contest/375842#problem/A

---

题意：

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

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

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

---

样例

//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
//Output
light the wand
accio
what?
what?

思路：

字符串哈希
将每一个字符串都转换为一个数值。例如：

原字符串：𝑎𝑒𝑑
𝐻𝑎𝑠ℎ1 = (1 ∗ 𝑠𝑒𝑒𝑑 ³ + 5 ∗ 𝑠𝑒𝑒𝑑 ² + 4 ∗ 𝑠𝑒𝑒𝑑 ¹ )% 𝑚𝑜𝑑。
代码中，1,5，4等系数用字符的ASCII码代替。

定义两个数组ss1,ss2，对于一对魔咒和功能，ss1中存魔咒，ss2中存功能。
对于每一条测试用例，先根据第一个字符判断是魔咒还是功能，再在对应的数组中找到其对应下标，再在另一个数组中得到匹配的功能和魔咒。

---

代码：

#include <iostream>
#include <string>
#include <map>
#include <cmath>
using namespace std;
const string t_end="@END@";
const double seed=7;
const int mod=1e9+7;
string s1,s2; 
map<int,int> mp1,mp2;
int hash1,hash2,idx1,idx2,n;
string ss1[100010],ss2[100010];

int main()
{
	while(getline(cin,s1))
	{
		if(s1==t_end)
			break;
		//截
		int p=s1.find("]");
		s2=s1.substr(p+2);
		s1=s1.substr(1,p-1);
		for(int i=0;i<s1.size();i++)	//魔咒 
		{
			hash1+=(int)s1[i]*pow(seed,s1.size()-i);
			hash1=hash1%mod;
		}
		mp1[++idx1]=hash1;
		ss1[idx1]=s1;
		for(int i=0;i<s2.size();i++)	//魔咒 
		{
			hash2+=(int)s2[i]*pow(seed,s2.size()-i);
			hash2=hash2%mod;
		}
		mp2[++idx2]=hash2;
		ss2[idx2]=s2;
	}
	cin>>n;
	getchar();
	for(int i=1;i<=n;i++)
	{
		getline(cin,s1);
		//cout<<"s1"<<s1<<"\n";
		if(s1[0]=='[')	//魔咒
		{
			s1=s1.substr(1,s1.size()-2);
			int pos=-1;
			for(int i=1;i<=idx1;i++)
			{
				if(ss1[i]==s1)
				{
					pos=i;
					break;
				}
			}
			if(pos!=-1)
					cout<<ss2[pos]<<"\n";	
				else
					cout<<"what?\n";
		}
		else
		{
			int pos=-1;
			for(int i=1;i<=idx2;i++)
			{
				if(ss2[i]==s1)
				{
					pos=i;
					break;
				}
			}
			if(pos!=-1)
					cout<<ss1[pos]<<"\n";	
				else
					cout<<"what?\n";
		}
		
	}
	return 0;
}

---