[백준] 14719번: 빗물 / C++

✏️ 문제 링크

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

14719번: 빗물

 

 

✏️ 문제 설명 (더보기 클릭 👆🏻)

문제

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. block[i]를 기준으로 왼쪽에서 가장 높은 블럭, 오른쪽에서 가장 높은 블럭을 찾는다. 

2. 두 블럭 중 더 작은 블럭 - block[i]를 하면 현재 블럭위로 채워지는 빗물을 계산할 수 있다. 

 

✏️ 문제 코드

#include <bits/stdc++.h>

using namespace std;

int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    
    int in, h, w, answer=0; cin>>h>>w;
    
    vector<int> block;
    
    while(w--){
        cin>>in;
        block.push_back(in);
    }
    
    for(int i=1; i<block.size()-1; i++){
        int l=0, r=0;
        for(int j=i-1; j>=0 ; j--){
            l = max(l, block[j]);
        }
        for(int j=i+1; j<block.size(); j++){
            r = max(r, block[j]);
        }
        int m = min(l, r) - block[i];
        if( m >= 0) answer+=m;
    }
    cout<<answer<<"\n";
    return 0;
}

 

 

 

 ⭐ if feedback and question : comment please⭐