백준 16360 Go Latin

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

16360번: Go Latin

 

def sol(s):
    pseudo_latin = {'a':'as',
                    'i':'ios',
                    'y':'ios',
                    'l':'les',
                    'n':'anes',
                    'ne':'anes',
                    'o':'os',
                    'r':'res',
                    't':'tas',
                    'u':'us',
                    'v':'ves',
                    'w':'was'}
    if s[-1] in pseudo_latin:
        s=s[:len(s)-1]+pseudo_latin[s[-1]]
    elif s[len(s)-2:] in pseudo_latin:
        s = s[:len(s)-2] + pseudo_latin[s[len(s)-2:]]
    else:
        s = s+'us'
 
    return s
 
if __name__ == '__main__':
    n = int(input())
    for _ in range(n):
        s = input()
        print(sol(s))

속도 제한이 있어서 딕셔너리에 접미사를 두고 풀었씁니다