[Baekjoon/Python] 1871번 - 좋은 자동차 번호판

[백준] Baekjoon Online Judge

문제로 이동

 

문제

앨버타의 자동차 번호판은 ABC-0123 (세 글자, 네 숫자)와 같이 두 부분으로 나누어져 있다.

좋은 번호판은 첫 번째 부분의 가치와 두 번째 부분의 가치의 차이가 100을 넘지 않는 번호판이다.

글자로 이루어진 첫 번째 부분의 가치는 글자를 26진법 수처럼 계산한다. (각 자리가 [A..Z]) 예를 들어, "ABC"의 가치는 28 (0×262 + 1×261 + 2×260)이 된다. "ABC-0123"은  |28 - 123| ≤ 100 이기 때문에, 좋은 번호판이다.

자동차 번호판이 주어졌을 때, 좋은 번호판인지 아닌지를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 번호판의 수 N (1 ≤ N ≤ 100)이 주어진다. 다음 N개 줄에는 자동차 번호판이 LLL-DDDD 형식으로 주어진다.

 

출력

각각의 자동차 번호판에 대해서, 좋은 번호판이면 "nice"를, 아니면 "not nice"를 출력한다.

 

---

풀이

N = int(input())

for i in range(N):
    a, b = input().split('-')
    
    num1, num2 = 0, int(b)
    
    for j in range(3):
        num1 += ((ord(a[2 - j]) - 65) * (26 ** j))
    
    if abs(num1 - num2) <= 100:
        print("nice")
    else:
        print("not nice")

 

ord()	문자를 아스키코드 숫자로 변환하는 함수
abs()	절대값을 구하는 함수
a ** b	a의 b 제곱

 위의 함수를 사용하여 문제를 풀었다.