[백준 알고리즘][C++] 이분 탐색(binary search) - 3020번 개똥벌레

안녕하세요~

​

이번시간에는 3020번 개똥벌레 문제를 풀어보려고 합니다.

 

www.acmicpc.net/problem/3020

3020번: 개똥벌레

 

 

명심하셔야 될 것은 일단 문제를 풀기 전에 코딩을 먼저 시작하지 마시고, 반드시 연습장으로 생각을 정리하고! 이 후에 생각이 맞는 것 같다고 확신이 되면 그 때 코딩을 시작하세요. 코드를 작성하면서 문제를 풀게되면 정리가 안된 상태기 때문에 코드 수정이 잦아지고, 정신이 없어지게 됩니다ㅠㅠ

그러니 반드시 노트에 어떻게 풀 것인지에 대한 정리를 다 하시고 코딩을 시작하시기 바랍니다!!

​

binary search 공부에 좋은 문제입니다.

 

문제

개똥벌레 한 마리가 장애물(석순과 종유석)로 가득찬 동굴에 들어갔다. 동굴의 길이는 N미터이고, 높이는 H미터이다. (N은 짝수) 첫 번째 장애물은 항상 석순이고, 그 다음에는 종유석과 석순이 번갈아가면서 등장한다.

아래 그림은 길이가 14미터이고 높이가 5미터인 동굴이다. (예제 그림)

이 개똥벌레는 장애물을 피하지 않는다. 자신이 지나갈 구간을 정한 다음 일직선으로 지나가면서 만나는 모든 장애물을 파괴한다.

위의 그림에서 4번째 구간으로 개똥벌레가 날아간다면 파괴해야하는 장애물의 수는 총 여덟개이다. (4번째 구간은 길이가 3인 석순과 길이가 4인 석순의 중간지점을 말한다)

하지만, 첫 번째 구간이나 다섯 번째 구간으로 날아간다면 개똥벌레는 장애물 일곱개만 파괴하면 된다.

동굴의 크기와 높이, 모든 장애물의 크기가 주어진다. 이때, 개똥벌레가 파괴해야하는 장애물의 최솟값과 그러한 구간이 총 몇 개 있는지 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N과 H가 주어진다. N은 항상 짝수이다. (2 ≤ N ≤ 200,000, 2 ≤ H ≤ 500,000)

다음 N개 줄에는 장애물의 크기가 순서대로 주어진다. 장애물의 크기는 H보다 작은 양수이다.

 

출력

첫째 줄에 개똥벌레가 파괴해야 하는 장애물의 최솟값과 그러한 구간의 수를 공백으로 구분하여 출력한다.

 

예제 입력 1

14 5
1
3
4
2
2
4
3
4
3
3
3
2
3
3

예제 출력 1

7 2

 

정답 코드

#include <iostream>
using namespace std;

int lower[500002];
int upper[500002];
int N, H;
int main()
{
	cin >> N >> H;
	int size;
	for (int i = 0; i <= H; i++)
	{
		lower[i] = 0;
		upper[i] = 0;
	}
	for (int i = 0; i < N; i++)
	{
		cin >> size;
		if (i % 2 == 0)
		{
			lower[size]++;
		}
		else
		{
			upper[size]++;
		}
	}
	for (int i = H; i > 1; i--)
	{
		lower[i - 1] = lower[i] + lower[i - 1];
		upper[i - 1] = upper[i] + upper[i - 1];
	}

	int total = 0;
	int minval = 987654321;
	int count = 0;
	for (int height = 1; height <= H; height++)
	{
		total = lower[height] + upper[H + 1 - height];
		if (total < minval)
		{
			minval = total;
			count = 1;
		}
		else if (total == minval)
		{
			count++;
		}
	}
	cout << minval << " " << count << endl;
}