[C/C++] 백준 3020번 - 개똥벌레

 

문제

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

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

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

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

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

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

입력

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

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

 

<코드>

#include<iostream>
#include<algorithm>
using namespace std;

int N, H, h;
long long stalagmite[500001], stalactite[500001]; // 석순 ,종유석
long long sum[500001];

int main() 
{
	ios::sync_with_stdio(0); 
	cin.tie(0); 

	cin >> N >> H;

	for (int i = 1; i <= N; i++)
	{
		cin >> h;
		if (i % 2 == 1) // 석순
		{
			stalagmite[h]++;
		}
		else // 종유석
		{
			stalactite[h]++;
		}
	}

	for (int i = H - 2; i >= 1; i--) // 석순, 종유석 누적 합
	{
		stalagmite[i] += stalagmite[i + 1];
		stalactite[i] += stalactite[i + 1];
	}

	sum[1] = stalagmite[1];
	sum[H] = stalactite[1];

	for (int i = 2; i <= H - 1; i++)
	{
		sum[i] = stalagmite[i] + stalactite[H + 1 - i];
	}

	/*
	for (int i = H; i >= 1; i--)
	{
		cout << "높이 "<<i <<"일때 파괴하는 장애물 수 : " << sum[i] << endl;
	}
	*/
	
	sort(sum + 1, sum + H + 1);

	cout << sum[1] << " ";
	cout << upper_bound(sum + 1, sum + H + 1, sum[1]) - lower_bound(sum + 1, sum + H + 1, sum[1]);
}

 

 

 

 

www.acmicpc.net/problem/3020