# 構(gòu)造字典，'A' --> 0 ...
def char_2_num(x):
  list_s = []
  list_num = []
  for i in range(26):
    c = chr(i + 65)
    list_s.append(c)
    list_num.append(i)
  c_2_n = dict(map(lambda x, y: [x, y], list_s, list_num))
  return c_2_n[f'{x}']


# 構(gòu)造字典，0 ---> 'A' ...
def num_2_char(x):
  list_s = []
  list_num = []
  for i in range(26):
    c = chr(i + 65)
    list_s.append(c)
    list_num.append(i)
  n_2_c = dict(map(lambda x, y: [x, y], list_num, list_s))
  print(n_2_c[x],end='')

# 編碼
def encode():
  s = input('輸入需要編碼的字符： ')
  print('編碼后的結(jié)果為： ',end='')
  for j in s:
    if j.isspace():
      print(' ',end='')
    else:
      ek = a * char_2_num(j) + b
      result = ek % 26
      num_2_char(result)

# 求模26下a的逆
def inv_(x):
  for inv_a in range(1,26,2):
    for j in range(27):
      if x * inv_a == 26 * j + 1:
        return inv_a

# 解碼
def decode():
  s = input('輸入需要解碼的字符： ')
  print('解碼后的結(jié)果為： ',end='')
  for j in s:
    if j.isspace():
      print(' ',end='')
    else:
      dk = inv_(a) * (char_2_num(j) - b)
      result = dk % 26
      num_2_char(result)

# 輸入指令
answer = input(f'請(qǐng)輸入所需的操作：編碼/E or 解碼/D: ')

# 輸入?yún)?shù)a，b
a = int(input('請(qǐng)輸入a：'))
b = int(input('請(qǐng)輸入b: '))

try:
  if answer.upper() == 'E':
    encode()
  elif answer.upper() =='D':
    decode()
  else:
    print('輸入錯(cuò)誤！')
except KeyError:
  print('請(qǐng)正確輸入大寫字母！')