[BOJ] 2037

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

2037번: 문자메시지

주어진 조건들만 맞춰서 구현하면 되는 문제.
같은 그룹인지를 찾는 방법을 개선 할 수도 있을 것 같음.

# BOJ 2037
ar = ['', '', 'ABC', 'DEF', 'GHI', 'JKL', 'MNO', 'PQRS', 'TUV', 'WXYZ']
def get_rept(ch):
    for i in range(2, 10):
        if ch in ar[i]:
            return (i, ar[i].find(ch) + 1)
p, w = map(int, input().split(' '))
s = input()
if len(s) < 2:
    print(p * get_rept(s[0]))
else:
    res = p * get_rept(s[0])[1]
    l = len(s)
    for i in range(1, l):
        if s[i] == ' ':
            res += p
            continue
        prev = get_rept(s[i - 1])
        curr = get_rept(s[i])
        res += p * curr[1]
        if not prev:
            continue
        if prev[0] == curr[0]:
            res += w
    print(res)