백준 14720 c++

enum 타입을 사용해 보고 싶어서 써봤다.

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

void input_data(vector<int>& milk)
{
	int N, temp;
	int i;

	cin >> N;
	for (i = 0; i < N; i++)
	{
		cin >> temp;
		milk.push_back(temp);
	}

	return;
}

void find_answer(vector<int>& milk)
{
	//딸기0 -> 초코1 -> 바나나2 -> 딸기0
	int i;
	enum current_milk {
		strawberry, //0
		choco, //1
		banana, //2
	};
	int current = strawberry;
	int count = 0;

	for (i = 0; i < milk.size(); i++)
	{
		if (milk[i] == current)
		{
			count++;
			current = (current + 1) % 3;
			//cout << "current : " << current << "\n";
			//cout << "count  : " << count << "\n";
		}
	}
	cout << count << "\n";

	return;
}

int main(void)
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	vector<int> milk;

	input_data(milk);
	find_answer(milk);

	return 0;
}