[Python]백준 - 다이얼(5622번)

문제출처 - https://www.acmicpc.net/problem/5622

5622번: 다이얼

내 풀이

string = list(input())

for i in range(len(string)):
    if string[i] == 'A' or string[i] == 'B' or string[i] == 'C':
        string[i] = 2
    elif string[i] == 'D' or string[i] == 'E' or string[i] == 'F':
        string[i] = 3
    elif string[i] == 'G' or string[i] == 'H' or string[i] == 'I':
        string[i] = 4
    elif string[i] == 'J' or string[i] == 'K' or string[i] == 'L':
        string[i] = 5
    elif string[i] == 'M' or string[i] == 'N' or string[i] == 'O':
        string[i] = 6
    elif string[i] == 'P' or string[i] == 'Q' or string[i] == 'R' or string[i] == 'S':
        string[i] = 7
    elif string[i] == 'T' or string[i] == 'U' or string[i] == 'V':
        string[i] = 8
    else:
        string[i] = 9

print(sum(string) + len(string))

나는 너무 더럽게 풀었는데

 

다른 풀이

dial = ['ABC', 'DEF', 'GHI', 'JKL', 'MNO', 'PQRS', 'TUV', 'WXYZ']
a = input()
ret = 0
for j in range(len(a)):
    for i in dial:
        if a[j] in i:
            ret += dial.index(i)+3
print(ret)

 

파이썬의 in 연산자를 활용하면 훨씬 간단하게 풀 수 있다.