[python] 14720, 우유 축제 : 구현, 그리디
14720번: 우유 축제
영학이는 딸기우유, 초코우유, 바나나우유를 좋아한다. 입맛이 매우 까다로운 영학이는 자신만의 우유를 마시는 규칙이 있다. 맨 처음에는 딸기우유를 한 팩 마신다. 딸기우유를 한 팩 마신 후
www.acmicpc.net
[문제]
영학이는 딸기우유, 초코우유, 바나나우유를 좋아한다.
입맛이 매우 까다로운 영학이는 자신만의 우유를 마시는 규칙이 있다.
1. 맨 처음에는 딸기우유를 한 팩 마신다.
2. 딸기우유를 한 팩 마신 후에는 초코우유를 한 팩 마신다.
3. 초코우유를 한 팩 마신 후에는 바나나우유를 한 팩 마신다.
4. 바나나우유를 한 팩 마신 후에는 딸기우유를 한 팩 마신다.
영학이는 우유 축제가 열리고 있는 우유거리에 왔다. 우유 거리에는 우유 가게들이 일렬로 늘어서 있다.
영학이는 우유 거리의 시작부터 끝까지 걸으면서 우유를 사먹고자 한다.
각각의 우유 가게는 딸기, 초코, 바나나 중 한 종류의 우유만을 취급한다.
각각의 우유 가게 앞에서, 영학이는 우유를 사마시거나, 사마시지 않는다.
우유거리에는 사람이 많기 때문에 한 번 지나친 우유 가게에는 다시 갈 수 없다.
영학이가 마실 수 있는 우유의 최대 개수를 구하여라.
[입력]
첫째 줄에 우유 가게의 수 N이 주어진다. (1 ≤ N ≤ 1000)
둘째 줄에는 우유 가게 정보가 우유 거리의 시작부터 끝까지 순서대로 N개의 정수로 주어진다.
0은 딸기우유만을 파는 가게, 1은 초코우유만을 파는 가게, 2는 바나나우유만을 파는 가게를 뜻하며, 0, 1, 2 외의 정수는 주어지지 않는다.
[출력]
영학이가 마실 수 있는 우유의 최대 개수를 출력하시오.
[예제]
[풀이]
딸기 → 초코 → 바나나 순으로 우유를 섭취하고 이를 0 → 1 → 2라고 숫자로 표현한다.
answer을 지금까지 먹은 우유의 개수로 설정하고, 이를 3으로 나누었을 때 나머지가 곧 이번에 먹어야 할 우유의 종류가 된다. 예를 들어, answer = 6이면 딸기 → 초코 → 바나나를 총 두번 반복 후 딸기를 먹어야 할 차례이다. 이때 지나가는 우유집에서 파는 우유가 딸기라면 answer에 +1을 해주는 방식으로 풀이하면 된다.
n = int(input())
milk = map(int, input().split())
answer = 0
for i in milk:
if i == answer % 3:
answer += 1
print(answer)