백준 그리디 알고리즘 14720 우유 축제

영학이는 딸기우유, 초코우유, 바나나우유를 좋아한다.

입맛이 매우 까다로운 영학이는 자신만의 우유를 마시는 규칙이 있다.

1. 맨 처음에는 딸기우유를 한 팩 마신다.
2. 딸기우유를 한 팩 마신 후에는 초코우유를 한 팩 마신다.
3. 초코우유를 한 팩 마신 후에는 바나나우유를 한 팩 마신다.
4. 바나나우유를 한 팩 마신 후에는 딸기우유를 한 팩 마신다. 

영학이는 우유 축제가 열리고 있는 우유거리에 왔다. 우유 거리에는 우유 가게들이 일렬로 늘어서 있다.

영학이는 우유 거리의 시작부터 끝까지 걸으면서 우유를 사먹고자 한다.

각각의 우유 가게는 딸기, 초코, 바나나 중 한 종류의 우유만을 취급한다.

각각의 우유 가게 앞에서, 영학이는 우유를 사마시거나, 사마시지 않는다.

우유거리에는 사람이 많기 때문에 한 번 지나친 우유 가게에는 다시 갈 수 없다.

영학이가 마실 수 있는 우유의 최대 개수를 구하여라.

입력

첫째 줄에 우유 가게의 수 N이 주어진다. (1 ≤ N ≤ 1000)

둘째 줄에는 우유 가게 정보가 우유 거리의 시작부터 끝까지 순서대로 N개의 정수로 주어진다.

0은 딸기우유만을 파는 가게, 1은 초코우유만을 파는 가게, 2는 바나나우유만을 파는 가게를 뜻하며, 0, 1, 2 외의 정수는 주어지지 않는다.

출력

영학이가 마실 수 있는 우유의 최대 개수를 출력하시오.

풀이

영학이는 딸기우유를 처음으로 먹고 딸기->초코->바나나->딸기 순으로 우유를 마신다.

입력값으로 보면 0 1 2 0순으로  마셔야한다.

딸기를 맨처음 먹어야 하므로 state는 0으로 초기화 한다.

그리고 입력값을 저장한 리스트를 순회하는데

리스트의 값이 state%3과 같으면 count를 증가시키고 0 다음 1, 1다음 2를 먹어야 하므로 state값도 1증가 시킨다.

state값은 0 1 2 3 4 5 이렇게 계속 증가하게 되는데 3으로 나눈 나머지로 비교하기 때문에

0 1 2 0 1 2를 반복하게 되어 요구하는것과 같은 비교를 할 수 있다. 

 

import sys
N = int(sys.stdin.readline())
inputs = list(map(int,sys.stdin.readline().rstrip("\n").split()))
state=0
count=0
for i in range(0,len(inputs)):
    if(inputs[i] == state%3):
       count+=1
       state+=1
print(count)