1일 1솔

[백준] 14720번: 우유 축제 (Python, 파이썬)

junmukbap98 2023. 9. 13. 15:34

문제 링크: https://www.acmicpc.net/problem/14720

 

14720번: 우유 축제

영학이는 딸기우유, 초코우유, 바나나우유를 좋아한다. 입맛이 매우 까다로운 영학이는 자신만의 우유를 마시는 규칙이 있다. 맨 처음에는 딸기우유를 한 팩 마신다. 딸기우유를 한 팩 마신 후

www.acmicpc.net

Key point

딸기 (0) -> 초코 (1) -> 바나나 (2) -> 딸기 (0) .. 순서로 우유를 마신다.

무조건 딸기를 먼저 마시므로, target = 0 으로 설정. 

그 다음 딸기를 마셨다면 (if milk[i] == target % 3:) res += 1 하고, 그 다음 target = 1 (초코) 로 설정한다. 

그 다음 초코를 마셨다면 (if milk[i] == target % 3:) res += 1 하고, 그 다음 target = 2 (바나나) 로 설정한다.

 

*여기서 바나나 (2) 를 마신 다음에 target이 다시 딸기 (0)로 바뀌기 때문에 target % 3을 사용한다!

n = int(input())
milk = list(map(int, input().split()))
target = 0
res = 0
for i in range(n):
    if milk[i] == (target % 3):
        res += 1
        target += 1

print(res)