HHUOJ 1796 魔咒词典

HHUOJ 1796 魔咒词典

题目描述

哈利波特在魔法学校的必修课之一就是学习魔咒。据说魔法世界有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?

这题看了一些网上的代码，都比较复杂，其实用两个map建立双向映射就很简单了，但是还是WA了两次，有两个坑点如下：
1.魔咒中也会带空格，这点样例中就有，要仔细！
2.输入可能为“[魔咒]”或“魔咒”，所以我统一以“魔咒”的形式存入
AC代码如下：

#include<bits/stdc++.h>
using namespace std;

int main()
{
    map<string,string>m1,m2;
    string ss,s,s1,s2;
    int n;
    while(getline(cin,ss) && ss!="@END@")
    {
        s1=ss.substr(1,ss.find(']',0)-1);//存入S1
        s2=ss.substr(ss.find(']',0)+2);//存入S2
        m1[s1]=s2;
        m2[s2]=s1;
    }
    cin>>n;
    getchar();
    while(n--)
    {
       getline(cin,s);
       if(s[0]=='[') s=s.substr(1,s.length()-2);//将“[魔咒]”转化为“魔咒”
       if(m1.count(s)!=0)
        cout<<m1[s]<<endl;
        else if(m2.count(s)!=0)
        {
           cout<<m2[s]<<endl;
        }
        else cout<<"what?"<<endl;
    }
    return 0;
}