백준 2304 - 창고 다각형

문제

 

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

2304번: 창고 다각형

 

 

풀이

 

가장 긴 막대 기둥을 기준으로 왼쪽과 오른쪽으로 나눈다. 왼쪽 부터 가장 긴 막대 기둥까지 진행하면서 스택의 최대 높이를 갱신하면서 넓이를 더해준다. 마찬가지로 오른쪽에서도 가장 긴 막대 기둥까지 진행하면서 스택의 최대 높이를 갱신하면서 넓이를 더해준다. 

 

 

코드

 

#include <iostream>
#include <stack>
 
using namespace std;
 
stack<int> st;
int N;
int arr[1001];
int pos, len;
int first = 1001, last, longest;
int result;
 
int main()
{
    cin.tie(0);
    ios::sync_with_stdio(0);
 
    cin >> N;
 
    while (N--)
    {
        cin >> pos >> len;
        arr[pos] = len;
        if (last < pos) last = pos;
        if (first > pos) first = pos;
        if (len > arr[longest]) longest = pos;
    }
 
    for (int i=first; i<=longest; i++)
    {
        if (arr[i])
        {
            if (st.empty()) st.push(arr[i]);
            else if(arr[i] > st.top()) st.push(arr[i]);
        }
        result += st.top();
    }
 
    while (!st.empty()) st.pop();
 
    for (int i=last; i>longest; i--)
    {
        if (arr[i])
        {
            if (st.empty()) st.push(arr[i]);
            else if(arr[i] > st.top()) st.push(arr[i]);
        }
        result += st.top();
    }
 
    cout << result;
 
    return 0;
}