[백준] 6198. 옥상정원 꾸미기(c++)

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

애먹어서 바로 리뷰한다..

입력값의 범위를 제대로 보지 않고 987654321를 최댓값으로 넣었다가 혼쭐이 났다. (경계값 분석이 역시나 중요하구나)

추가적으로 이전 코테 문제(https://www.acmicpc.net/problem/2493)에서 +된 부분은 마지막에 flush 해줘야 하는 부분이 달랐다.

 

---

#include<iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <string>
#include <map>
#include <climits>
#include <queue>
#include <set>
#include <cmath>
#include <stack>
#include <list>

using namespace std;

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

    int n;

    // 입력값 받을 배열
    int arr[80002];
    long long int answer = 0;
    // 스택
    stack<pair<int, int> > stk;

    cin >> n;

    stk.push(make_pair(1087654321, 0));

    for (int i = 1; i <= n; i++) {
        cin >> arr[i];
        while (stk.top().first <= arr[i]) {
            answer += i - stk.top().second - 1;
            stk.pop();
        }
        stk.push(make_pair(arr[i], i));
    }

    while (stk.top().first < 1087654321) {
        answer += n - stk.top().second;
        stk.pop();
    }

    cout << answer << " ";

    return 0;
}