[Baekjoon] 빗물

👀 문제 설명

문제

2차원 세계에 블록이 쌓여있다. 비가 오면 블록 사이에 빗물이 고인다.

비는 충분히 많이 온다. 고이는 빗물의 총량은 얼마일까?

 

입력

첫 번째 줄에는 2차원 세계의 세로 길이 H과 2차원 세계의 가로 길이 W가 주어진다. (1 ≤ H, W ≤ 500)

두 번째 줄에는 블록이 쌓인 높이를 의미하는 0이상 H이하의 정수가 2차원 세계의 맨 왼쪽 위치부터 차례대로 W개 주어진다.

따라서 블록 내부의 빈 공간이 생길 수 없다. 또 2차원 세계의 바닥은 항상 막혀있다고 가정하여도 좋다.

 

출력

2차원 세계에서는 한 칸의 용량은 1이다. 고이는 빗물의 총량을 출력하여라.

빗물이 전혀 고이지 않을 경우 0을 출력하여라.

 

예제 입력 1

4 4

3 0 1 4

 

예제 출력 1

5

 

예제 입력 2

4 8

3 1 2 3 4 1 1 2

 

예제 출력 2

5

 

예제 입력 3

3 5

0 0 0 2 0

 

예제 출력 3

0

 

✍🏻풀이

현재 위치를 기준으로 왼쪽에서 가장 큰 것(1)과, 오른쪽에서 가장 큰 것(2)을 구한다.

(1)과 (2) 중 더 작은 것에서 현재 위치의 값을 빼면 현재 위치에서 받을 수 있는 물의 양을 구할 수 있다.
(NHN인가,, 카카오에 이것과 비슷한 문제가 나왔던 것 같다.)

 

코드

//
//  BOJ_14719.cpp
//  Algorithm
//
//  Created by 조수민 on 2020/10/25.
//  Copyright © 2020 조수민. All rights reserved.
//
//  빗물(https://www.acmicpc.net/problem/14719)

#include <stdio.h>
#include <iostream>
#include <vector>

using namespace std;

int main() {
    int H, W;
    int answer = 0;
    
    cin >> H >> W;
    vector<int> v(W);
    
    // input
    for (int i = 0; i < W; i++) {
        cin >> v.at(i);
    }
    
    for (int i = 1; i < W - 1; i++) {
        int left = 0;
        int right = 0;
        
        // 왼쪽 최대
        for (int j = 0; j < i; j++) {
            left = max(left, v.at(j));
        }
        
        // 오른쪽 최대
        for (int j = W - 1; j > i; j--) {
            right = max(right, v.at(j));
        }
        
        answer += max(0, min(left, right) - v[i]);
    }
    
    cout << answer << endl;
    return 0;
}