乙级 1076 Wifi密码

下面是微博上流传的一张照片：“各位亲爱的同学们，鉴于大家有时需要使用 wifi，又怕耽误亲们的学习，现将 wifi 密码设置为下列数学题答案：A-1；B-2；C-3；D-4；请同学们自己作答，每两日一换。谢谢合作！！~”—— 老师们为了促进学生学习也是拼了…… 本题就要求你写程序把一系列题目的答案按照卷子上给出的对应关系翻译成 wifi 的密码。这里简单假设每道选择题都有 4 个选项，有且只有 1 个正确答案。

输入格式：

输入第一行给出一个正整数 N（≤?100），随后 N 行，每行按照?编号-答案?的格式给出一道题的 4 个选项，T?表示正确选项，F?表示错误选项。选项间用空格分隔。

输出格式：

在一行中输出 wifi 密码。

输入样例：

8
A-T B-F C-F D-F
C-T B-F A-F D-F
A-F D-F C-F B-T
B-T A-F C-F D-F
B-F D-T A-F C-F
A-T C-F B-F D-F
D-T B-F C-F A-F
C-T A-F B-F D-F

输出样例：

13224143

?下面是我的代码：

#include <bits/stdc++.h>            //万能头文件
using namespace std;
int main()
{
    int n;    cin >> n;
    string s;
    getline(cin,s);                    //读掉n后面的回车
    
    char temp;                        //记录读到的字母
    string ans;                       //记录符合要求的字母
    
    map<char,int> mp;                //照题目要求，一个字母对应一个数字，不这样写的话用if也行
    mp['A'] = 1;
    mp['B'] = 2;
    mp['C'] = 3;
    mp['D'] = 4;
    
    for (int i = 0; i < n; i++) {                //输入n个字符串
        getline(cin,s);
        for (int j = 0; j < s.size(); j++) {                //遍历字符串
            if (s[j] >= 'A' && s[j] <= 'D') {               //只要读到字母，就把字母存为temp，然后等待后面T或F的判断，如果是T，就会被存进答案字符串，如果是F，则temp会被下一个字母覆盖
                temp = s[j];
            }
            if (s[j] == 'T') {
                ans += temp;
                break;
            }
        }
    }

    for (int i = 0; i < n; i++) {            //将字母都转化为数字，输出
        cout << mp[ans[i]];
    }
}