[백준] 14719번: 빗물(JAVA)

▶ 문제

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

14719번: 빗물

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

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

입력

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

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

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

출력

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

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

---

▶ 풀이

 - JAVA

import java.io.IOException;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main{
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int H = Integer.parseInt(st.nextToken());
        int W = Integer.parseInt(st.nextToken());

        st = new StringTokenizer(br.readLine());
        int[] block = new int[W];
        for(int i = 0; i < W; i++)
            block[i] = Integer.parseInt(st.nextToken());

        int sum = 0;
        for(int i = 1; i <= W-2; i++){
            int l = 0, r = 0;
            for(int j = 0; j < i; j++)
                l = Math.max(l, block[j]);
            for(int j = W-1; j > i; j--)
                r = Math.max(r, block[j]);

            int tmp = Math.min(l,r)-block[i];
            sum += (tmp>0)? tmp : 0;
        }

        System.out.print(sum);
    }
}

---

▶ 메모

2차원 세계의 제일 좌측과 우측은 블록이 쌓여야만 빗물이 고이므로 합 계산 for 문에서 제외한다.
좌측과 우측 모두 현 위치보다 높아야 빗물이 쌓이므로 해당 부분을 고려하여 합 계산 해준다.