백준 12778 CTP공국으로 이민 가자

https://www.acmicpc.net/problem/12778

12778번: CTP공국으로 이민 가자

 

def sol(type_prob, probs):
    if type_prob == 'C':
        answer = [ord(i)-64 for i in probs]
    elif type_prob == 'N':
        answer = [chr(int(i) + 64) for i in probs]
 
    return answer
 
if __name__ == '__main__':
    t = int(input())
    for _ in range(t):
        m, type_prob = input().split()
        probs = list(input().split())
        print(*sol(type_prob, probs))
 

'A' 의 아스키코드 값은 65라는 사실

Ascii code of A is 65