[백준 / C++] 과일 탕후루 (실버2, 30804)

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

문제

입력

첫 줄에 과일의 개수 N이 주어집니다. (1≤N≤200000)

둘째 줄에 탕후루에 꽂힌 과일을 의미하는 N개의 정수 S1,, SN이 공백으로 구분되어 주어집니다. (1≤Si≤9)

출력

문제의 방법대로 만들 수 있는 과일을 두 종류 이하로 사용한 탕후루 중에서, 과일의 개수가 가장 많은 탕후루의 과일 개수를 첫째 줄에 출력하세요.

예제 입력 1 

5
5 1 1 2 1

예제 출력 1 

4

과일을 앞에서 1개, 뒤에서 0개의 과일을 빼면 남은 과일은 1, 1, 2, 1번 과일이 꽂혀있는 탕후루가 됩니다. 과일의 개수는 4개입니다.

예제 입력 2 

3
1 1 1

예제 출력 2 

3

탕후루가 이미 두 종류 이하의 과일로만 이루어져 있습니다.

예제 입력 3 

9
1 2 3 4 5 6 7 8 9

예제 출력 3 

2

과일을 앞에서 3$3$개, 뒤에서 4$4$개의 과일을 빼면 남은 과일은 4,5번 과일이 꽂혀있는 탕후루가 됩니다. 과일의 개수는 2개입니다.

---

문제 풀이

이번 문제는 투 포인터를 활용해 2 종류의 과일만을 사용하여 과일의 개수가 가장 많은 탕후루를 만드는 문제입니다.

 

문제에서 미리 주어진 배열을 바탕으로 앞과 뒤에서 몇 개의 과일을 빼는지 체크하기 위해 처음에는 완전탐색으로 접근했습니다.

 

중복된 과일의 종류와 수를 map으로 저장해두고 왼쪽에서 뺀 경우와 오른쪽에서 뺀 경우로 분기시켜 완전탐색하였으며 시간 초과 결과를 받게 되었습니다.

 

이후에는 DP를 사용해 최적화를 하려고 시도해보았지만 비슷한 결과물을 받고 문제 풀이 방식을 수정하게 되었으며 투 포인터를 사용하게 되었습니다.

 

다음과 같은 배열이 있다고 예시를 들어보도록 하겠습니다.

arr : 2 4 2 2 1 5 7 9 (size : 8)

 

2개의 인덱스를 사용해 순환하기 위해 다음과 같이 초기화합니다.

현재는 2, 4 두개의 종류의 과일을 사용하고 있으며, 올바른 상태이므로 R 인덱스를 오른쪽으로 1씩 증가시켜 갑니다.

R인덱스를 오른쪽으로 이동시키다 올바르지 않은 상태(2, 4, 1 - 3개의 과일을 사용)가 되었으므로 L인덱스를 올바를 때까지 이동시킵니다.

이러한 과정을 반복하며 올바른 상태의 최대 크기를 구하는 방식으로 문제를 해결하게 되었습니다.

(종류별 개수는 map 컨테이너로 관리)

코드

#include <iostream>
#include <vector>
#include <map>

using namespace std;

int n;
vector<int> v;
map<int, int> m;

int mx = -1;
int arr[10];

int main()
{
	// input
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		int temp = 0;
		cin >> temp;

		v.push_back(temp);
	}

	int lIdx = 0;
	int rIdx = 1;

	m[v[lIdx]]++;
	m[v[rIdx]]++;

	int cnt = m.size();

	while (true)
	{
		if (lIdx >= rIdx) break;

		// 만약 현재 카운트가 2 이하라면 rIdx 증가
		if (cnt <= 2 && rIdx < v.size())
		{
			rIdx++;

			if (rIdx >= v.size())
			{
				continue;
			}

			if (m.find(v[rIdx]) == m.end())
			{
				m.insert({ v[rIdx], 1 });
				cnt++;
			}
			else if (m[v[rIdx]] == 0)
			{
				m[v[rIdx]]++;
				cnt++;
			}
			else
			{
				m[v[rIdx]]++;
			}
		}
		// 아니라면 lIdx 증가
		else
		{
			// 현재 요소의 수 계산
			mx = max(mx, rIdx - lIdx);

			m[v[lIdx]]--;
			if (m[v[lIdx]] == 0) cnt--;
			lIdx++;
		}		
	}

	cout << mx;
}