【Week 15 作业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?

思路

字符串哈希

将字符串转化为一个数字以便于快速比较。

如，字符串aed
Hash值：(1*seed³+5seed²+4seed¹)%mod;
seed 常取值为7、17、131等质数
mod常见取值1e9+7或采用unsigned long long 自然溢出

本题应用

将魔咒与对应功能保存到数组中，使用map<unsigned long long,int>来记录字符串哈希值与数组下标。
对于要查询的字符串，求取其哈希值并在map中查找，若存在则输出对应下标的字符串，否则输出“what?”。

代码

#include <iostream>
#include <string.h>
#include <map>
using namespace std;
const int seed=7; 
char str1[100005][25];
char str2[100005][85];
map<unsigned long long,int> mp;
unsigned long long Hash(char*str,int l,int r)
{
	unsigned long long ans=0,temp=seed;
	for(int i=l;i<=r;i++)
	{
		if(str[i]!=' ')
		{
			ans+=str[i]*temp;
			temp*=temp;
		}
	}
	return ans;
}
int main(int argc, char** argv) {
	int n=0;
	char temp[105];
	while(scanf("%[^\n]",temp)&&strcmp(temp,"@END@")!=0)
	{
		getchar();
		int len=strlen(temp);
		int l=1,r=1;
		for(;r<len&&temp[r]!=']';r++);
		strncpy(str1[n],temp+1,r-l);
		unsigned long long h=Hash(temp,l,r-1);
		mp[h]=n;
		l=r+2;r=len;
		strncpy(str2[n],temp+l,r-l);
		h=Hash(temp,l,r-1);
		mp[h]=n;
		n++;
	}
	int num;
	scanf("%d",&num);getchar();
	for(int i=0;i<num;i++)
	{
		scanf("%[^\n]",temp);getchar();
		int len=strlen(temp);
		unsigned long long h;
		if(temp[0]=='[')
			h=Hash(temp,1,len-2);
		else
			h=Hash(temp,0,len-1);
		if(mp.find(h)==mp.end())
			printf("what?\n");
		else
		{
			if(temp[0]=='[')
				printf("%s\n",str2[mp[h]]);
			else
				printf("%s\n",str1[mp[h]]);
		}
	}
	return 0;
}