PTA(Basic Level)-1076 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

二 代码

# -*- coding:UTF-8 -*-

num = int(input())
options = {chr(ord('A')+i): i+1 for i in range(4)}
codes = []
while (num > 0):
    num = num - 1
    ansline = input()
    pos = ansline.find("T")
    result_char = ansline[pos-2]
    # 输出对应wifi密码位
    if result_char in options:
        codes.append(options[result_char])

codes = list(map(str, codes))
wifi_code = "".join(codes)      
print(wifi_code)