BOJ 14719 : 빗물

• 문제 링크 : boj.kr/14719
• 난이도 : G5
• 태그 : 구현

14719번: 빗물

---

코드

#include <bits/stdc++.h>

#define all(x) (x).begin(), (x).end()

#define INF 0x7FFFFFFF

using namespace std;

using ll = long long;
using ld = long double;
using pii = pair<int,int>;
using pll = pair<ll, ll>;

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

    int h, w;
    cin >> h >> w;
    vector<int> arr(w);
    for(auto &u : arr) cin >> u;

    int ans = 0;
    for(int i = 1; i < w-1; i++) {
        int l = 0, r = 0;
        for(int j = 0; j < i; j++) {
            l = max(l, arr[j]);
        }
        for(int j = i+1; j < w; j++) {
            r = max(r, arr[j]);
        }
        ans += max(min(r,l) - arr[i], 0);
    }

    cout << ans;

    return 0;
}

---

풀이

어떤 세로줄에는 min(왼쪽 중 가장 높은 높이, 오른쪽 중 가장 높은 높이) - 그 세로줄의 높이 만큼 물이 쌓인다.

이를 O(n^2)으로 그대로 구현해주면 정답을 받을 수 있다.