[Python] 백준 14720번_우유 축제

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

14720번: 우유 축제

 

• 영학이는 자신만의 우유를 마시는 규칙이 있음
  • 1) 맨 처음에는 딸기우유를 한 팩 마심
  • 2) 딸기우유를 한 팩 마신 후에는 초코우유를 한 팩 마심
  • 3) 초코우유를 한 팩 마신 후에는 바나나우유를 한 팩 마심
  • 4) 바나나우유를 한 팩 마신 후에는 딸기우유를 한 팩 마심
• 우유 거리에는 우유 가게들이 일렬로 늘어서 있음
• 영학이는 우유 거리의 시작부터 끝까지 걸으면서 우유를 사먹고자 함
• 각각의 우유 가게는 딸기, 초코, 바나나 중 한 종류의 우유만을 취급
• 각각의 우유 가게 앞에서, 영학이는 우유를 사마시거나, 사마시지 않음
• 우유거리에는 사람이 많기 때문에 한 번 지나친 우유 가게에는 다시 갈 수 없음
• 영학이가 마실 수 있는 우유의 최대 개수 구하기

• 입력
  • 첫째 줄에 우유 가게의 수 N이 주어짐 (1 ≤ N ≤ 1000)
  • 둘째 줄에는 우유 가게 정보가 우유 거리의 시작부터 끝까지 순서대로 N개의 정수로 주어짐
  • 0은 딸기우유만을 파는 가게, 1은 초코우유만을 파는 가게, 2는 바나나우유만을 파는 가게를 뜻하며, 0, 1, 2 외의 정수는 주어지지 않음
• 출력
  • 영학이가 마실 수 있는 우유의 최대 개수를 출력
• 시간 제한: 1초
• 메모리 제한: 256 MB

 

"""
입출력 예시)

7
0 1 2 0 1 2 0
            -> 7
"""

 

 

## 의사코드 ##

# 맨 처음 마실 우유는 딸기우유(0)
# 그 다음부터 초코우유(1) -> 바나나우유(2) -> 딸기우유(0) 순서대로 마심

# 0 -> 1 -> 2 순서
# => 3으로 나눈 나머지 

# 현재 우유가 마실 수 있는 우유의 개수를 3으로 나눈 나머지와 같으면 
# 우유 개수 + 1

 

 

 

통과한 코드

 

n = int(input())
stores = list(map(int, input().split()))
cnt = 0 # 마실 수 있는 우유의 개수
for s in stores:
    # 현재 우유가 마실 수 있는 우유의 개수를 3으로 나눈 나머지와 같으면 우유 개수 + 1
    if s == cnt % 3:
        cnt += 1
print(cnt)

-> 마실 수 있는 우유는 0 - 1 - 2의 순서대로 반복된다는 점에서 지금까지 마신 우유의 개수(cnt)를 3으로 나눈 나머지로 현재 우유가 마실 수 있는 우유의 순서인지 확인

 

 

 

 

 

참고)

 

https://s0ng.tistory.com/entry/%EB%B0%B1%EC%A4%80-%EA%B7%B8%EB%A6%AC%EB%94%94-%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%EC%9A%B0%EC%9C%A0-%EC%B6%95%EC%A0%9C-14720%EB%B2%88-%ED%8C%8C%EC%9D%B4%EC%8D%AC-python

[백준] 그리디 알고리즘 - 우유 축제 (14720번) #파이썬 #python

 

https://velog.io/@suhado/%EB%B0%B1%EC%A4%80-14720-%EC%9A%B0%EC%9C%A0-%EC%B6%95%EC%A0%9C-python-%ED%8C%8C%EC%9D%B4%EC%8D%AC

[백준 14720: 우유 축제 ] python 파이썬