백준 14720번 우유축제(Python, C++)

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

14720번: 우유 축제

딸기(0) -> 초코(1) -> 바나나(2) -> 딸기(0) 순으로 우유를 먹어야 하기 때문에 특정 변수를 통해 다음에 어떤 우유를 마셔야 하는지 저장해놓은 다음 현재 우유와 비교하여 같은 때만 1씩 증가시켜 최대 우유를 얼마나 마실 수 있는지 구하면 된다. 

 

<python>

n = int(input())
c = list(map(int, input().split()))
max = 0

for i in range(n):
    if(c[i] == max%3):
        max+=1

print(max)

 

<C++>

#include <iostream>
using namespace std;

int main() {

	int n;
	cin >> n;
	int m[1000];
	int max = 0;
	int c = 0;

	for (int i = 0; i < n; i++) {
		cin >> m[i];
	}
	for (int j = 0; j < n; j++) {
		if (m[j] == 0 && c == 0) {
			max++;
			c = 1;
		}
		if (m[j] == 1 && c == 1) {
			max++;
			c = 2;
		}
		if (m[j] == 2 && c == 2) {
			max++;
			c = 0;
		}
	}

	cout << max;
	return 0;
}