다이얼 문제

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

def solution(inp):
    sec = 0
    dial = {'A':2, 'B':2, 'C':2, 'D':3, 'E':3, 'F':3, 'G':4, 'H':4, 'I':4, 'J':5,'K':5, 'L':5, 'M':6, 'N':6, 'O':6, 'P':7,'Q':7, 'R':7, 'S':7, 'T':8, 'U':8, 'V':8, 'W':9, 'X':9, 'Y':9, 'Z':9}
    for s in inp:
        num = dial[s]
        sec += (num+1)
    return sec

inp = input().upper()
print(solution(inp))

python dictionary 사용

 

def solution(inp):
    dial = ['ABC', 'DEF', 'GHI', 'JKL', 'MNO', 'PQRS', 'TUV', 'WXYZ']
    sec = 0
    for s in inp:
        for idx in range(len(dial)):
            if s in dial[idx]:
                sec += (idx+3)
    return sec

inp = input().upper()
print(solution(inp))

더 간단하게 python list 를 사용하는 방법이 있었다.