취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 2304번 문제인 창고 다각형이다.
문제는 아래 링크를 확인하자.

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

창고 다각형은 가장 높은 높이의 기둥 H값을 기준으로 다음과 같은 세 구역으로 나눌 수 있다.

1) 처음으로 높이가 H인 기둥이 나오는 곳을 기준으로 왼쪽

2) 마지막으로 높이가 H인 기둥이 나오는 곳을 기준으로 오른쪽

3) 1과 2 사이

1과 3의 경우, 창고다각형의 끝쪽에서 안쪽 방향으로 각 기둥을 살펴보면서 점점 더 높은 기둥의 높이를 앞으로의 높이로 계산해나가는 것으로 창고다각형의 최소 넓이를 구할 수 있다.

2의 경우, 그냥 두 기둥 사이의 거리와 H값을 이용해 최소 넓이를 구할 수 있다.

각각의 부분의 넓이를 구하고 더해 문제를 해결하자.

아래는 제출한 소스코드이다.

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

int mxH = 0;;
vector<pair<int, int>> vec;

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

	int N; cin >> N;
	for (int i = 0; i < N; i++) {
		int l, h; cin >> l >> h;
		if (h > mxH) mxH = h;
		vec.emplace_back(make_pair(l, h));
	}

	sort(vec.begin(), vec.end());

	int ans = mxH;
	int LL, RR;
	int oldH = 0, oldX = 0;
	for (auto& p : vec) {
		int x = p.first, h = p.second;
		ans += oldH * (x - oldX);
		oldX = x, oldH = max(oldH, h);
		if (oldH == mxH) {
			LL = x;
			break;
		}
	}
	reverse(vec.begin(), vec.end());
	oldX = 1000000007, oldH = 0;
	for (auto& p : vec) {
		int x = p.first, h = p.second;
		ans += oldH * (oldX - x);
		oldX = x, oldH = max(oldH, h);
		if (oldH == mxH) {
			RR = x;
			break;
		}
	}
	ans += oldH * (RR - LL);

	cout << ans;
}