본문 바로가기
코딩테스트 연습/백준 Boj

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

by Lagooni 2021. 11. 24.

문제

상도시의 차량 번호판 형식이 주어졌을 때, 가능한 차량 번호판의 개수를 구해보자.

  • 번호판에 사용할 수 있는 숫자는 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번 연속해서 불가능하다.

입력

첫째 줄에 차량 번호판의 형식이 주어진다. 형식은 길이가 4보다 작거나 같으며, c와 d로만 이루어져 있다.

출력

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

예제 입력 1

dd

예제 출력 1

90

00부터 99까지 총 100가지 중에서 00, 11, 22, 33, 44, 55, 66, 77, 88, 99가 불가능하다.

예제 입력 2

cc

예제 출력 2

650

262 = 676가지 중에서 같은 문자가 중복되는 26가지가 불가능하다.

예제 입력 3

dcdd

예제 출력 3

23400

풀이 방법

  • c 는 문자 (a~z 26가지 경우) d 는 숫자 (0~9 10가지 경우)
    • cc일 때 26*25가지 경우 (처음 뽑을 때 26개 * 처음 뽑은 것을 제외한 25개)
    • ccc일 때 26*25*25 가지 경우 (처음 뽑을 때 26개 * 처음 뽑은 것을 제외한 25개 * 두번째 뽑은 것을 제외한 25개)
    • ccc일 때 26*25*25*25 가지 경우 (처음 뽑을 때 26개 * 처음 뽑은 것을 제외한 25개 * 두번째 뽑은 것을 제외한 25개 * 세번째 뽑은 것을 제외한 25개)
  • d 도 똑같이 해준다.
import sys

n = list(sys.stdin.readline().rstrip())

answer = 1

if n[0] == 'd':
    answer = 10
else:
    answer = 26

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


print(answer)

댓글