백준 알고리즘 10551번 STROJOPIS(python)

영역을 직접 나눠서 리스트에 추가했다. 쉬운데 조금 귀찮은 문제다.

li = ['1QAZ', '2WSX', '3EDC', '4RFV5TGB', '6YHN7UJM', '8IK,', '9OL.', "0P;/-['=]"]
ans = [0]*8
for c in input():
    for i in range(8):
        if c in li[i]:
            ans[i] += 1
for n in ans:
    print(n)