[Python] 백준 16968번 차량 번호판 1

문제

상도시의 차량 번호판 형식이 주어졌을 때, 가능한 차량 번호판의 개수를 구해보자.
번호판에 사용할 수 있는 숫자는 0, 1, 2, ..., 8, 9이다.
사용할 수 있는 문자는 a, b, c, d, ..., y, z이다.
차량 번호판의 형식은 최대 4글자이고, c와 d로 이루어진 문자열로 나타낼 수 있다.
c는 문자가 위치하는 자리, d는 숫자가 위치하는 자리이다.
같은 문자 또는 숫자가 연속해서 2번 나타나면 안 된다.

예를 들어, 형식이 "cd"이면, a1, d4, h5, k4 등이 가능하다. 형식이 "dd"인 경우에 01, 10, 34, 69는 가능하지만, 00, 11, 55, 66은 같은 숫자가 2번 연속해서 불가능하다.

 

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

입력

첫째 줄에 차량 번호판의 형식이 주어진다. 

형식은 길이가 4보다 작거나 같으며, c와 d로만 이루어져 있다.

출력

첫째 줄에 가능한 차량 번호판의 개수를 출력한다.

예제

# input
dd
# output
90

# input
cc
# output
650

# input
dcdd
# output
23400

문제 풀이

P = input()

if P[0] == 'd':
    answer = 10
else:
    answer = 26
    
for i in range(1, len(P)):
    if P[i] == 'd':
        if P[i-1] == 'd':
            answer *= 9
        else:
            answer *= 10
    
    else:
        if P[i-1] == 'c':
            answer *= 25
        else:
            answer *= 26
            
print(answer)

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.