题目描述

A：最近出了一个饮料营养等级你们知道吗？例如无糖的饮料是 A 级，可乐是 D 级……
B：那……无糖可乐是什么级别？
C：AD 级吧。
A：出院！
B：出什么院，你也给我进去！

以上是某群中一段有趣的对话。请你按照里面的逻辑，在已知某些饮料的等级的情况下，给饮料定级。定级的方法是：

• 如果是已知等级的饮料，直接输出等级；
• 对于一个新饮料的名字，你需要将名字拆成两个已知等级的部分，然后输出这个级别。例如：Diet是A，Coke是D，那么DietCoke就是AD；
• 如果新饮料无法拆解或者有多种拆解方法，统一定为 D 级。

输入格式

输入第一行是两个正整数 N,M (1 \le N,M \le 100) ，表示已知的饮料有 N 种，需要定级的饮料有 M 种。

接下来首先是 N 行，每行是一个字符串和一个字符，表示一种饮料的名字和对应的等级，等级只有 A,B,C,D 四种。

然后是 M 行，每行是一个字符串，表示需要定级的饮料的名字。

所有饮料名字只包含有大小写字母，长度不超过 30 ，给定拥有等级的饮料的名字不会重复。

输出格式

对于每一个需要定级的饮料，输出定好的定级。

输入样例

5 6
Diet A
LowSugarTea B
Milk C
Coke D
Water A
DietCoke
Pepsi
Milk
CokeWater
GoodMilk
dietCoke

输出样例

AD
D
C
DA
D
D

题意

在已知某些饮料的等级的情况下，给饮料定级。定级的方法是：

• 如果是已知等级的饮料，直接输出等级；
• 对于一个新饮料的名字，你需要将名字拆成两个已知等级的部分，然后输出这个级别。例如：Diet是A，Coke是D，那么DietCoke就是AD；
• 如果新饮料无法拆解或者有多种拆解方法，统一定为 D 级。

思路

用 unordered_map 记录已知的饮料等级，枚举分割点，判断是否可拆成两个已知部分，最后按照题意要求输出结果。

代码

void solve(){
    int n,m;
    cin >> n >> m;
    unordered_map<string,string> mp;
    for(int i = 0;i < n;i++){
        string s1,s2;
        cin >> s1 >> s2;
        mp[s1] = s2;
    }
    for(int i = 0;i < m;i++){
        string s;
        cin >> s;
        if(mp.find(s) != mp.end()){
            cout << mp[s] << "\n";
        }else{
            int cnt = 0;
            string ans;
            for(int i = 0;i < s.size();i++){
                string s1 = s.substr(0,i + 1);
                string s2 = s.substr(i + 1);
                if(mp.find(s1) != mp.end() && mp.find(s2) != mp.end()){
                    ans = mp[s1] + mp[s2];
                    cnt++;
                }
            }
            if(cnt == 1){
                cout << ans << "\n";
            }else{
                cout << "D\n";
            }
        }
    }
}