[백준] 5622 파이썬(python) : 다이얼 - (★)

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

5622번: 다이얼

 

 

알파벳은 26글자이고 입력값의 최대값은 15이다. 그러면 O(n²)로 풀어도 시간제한에 걸리지 않는다.

 

s = input()
alpha = [ [], ['A', 'B', 'C'], [ 'D', 'E', 'F'], [ 'G', 'H', 'I'],
 ['J', 'K', 'L'], ['M', 'N', 'O'], [ 'P', 'Q', 'R', 'S'],
  ['T', 'U', 'V'], [ 'W', 'X', 'Y', 'Z'] ]
res = 0
for i in s:
    for j in range(9):
        if i in alpha[j]:
            res += (j+2)
print(res)