백준 5622번

안녕하세요 ㅎㅎ
오늘은 백준 5622번 다이얼 문제를 가지고 왔습니다 !

이 문제는 문제가 어렵다기 보다는 알파벳 별로 if-elif-else를 사용해서 분기문을 만드는 것이 조금은 귀찮은...ㅎ 문제입니다. 저는 조금 더 깔끔한 코드를 위해 분기문 부분을 함수화 했습니다.

다음은 정답코드 입니다.

def cal_time(x):
    sum=0
    if(x=='A' or x=='B' or x=='C'):
        sum =3
    elif(x=='D' or x=='E' or x=='F'):
        sum =4
    elif(x=='G' or x=='H' or x=='I'):
        sum =5
    elif(x=='J' or x=='K' or x=='L'):
        sum =6
    elif(x=='M' or x=='N' or x=='O'):
        sum =7
    elif(x=='P' or x=='Q' or x=='R' or x=='S'):
        sum =8
    elif(x=='T' or x=='U' or x=='V'):
        sum =9
    elif(x=='W' or x=='X' or x=='Y' or x=='Z'):
        sum =10
    return sum

arr = list(input())
result =0
for x in arr:
    a = cal_time(x)
    result +=a

print(result)