백준 14719번 : 빗물

문제

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

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

입력

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

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

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

출력

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

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

나의 풀이 (행 기준)

• 2차원 배열로 블록을 시각화한 뒤, 각 행마다 좌우 벽이 존재하는 경우, 그 사이에 있는 빈 칸의 개수를 빗물로 계산

import java.io.*;
import java.util.*;

/**
 * 백준 14719 빗물 (골드 5) - 구현
 * <a href="https://www.acmicpc.net/problem/14719">...</a>
 */
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        st = new StringTokenizer(br.readLine());

        int H = Integer.parseInt(st.nextToken());
        int W = Integer.parseInt(st.nextToken());

        int[][] blocks = new int[H][W];

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

            for (int j = 0; j < cnt; j++) {
                blocks[H - j - 1][i] = 1;
            }
        }
        int answer = 0;

        for (int i = 0; i < H; i++) {
            Integer preBlockIdx = null;
            for (int j = 0; j < W; j++) {
                if (blocks[i][j] == 1) {
                    if (preBlockIdx != null) {
                        answer += j - preBlockIdx - 1;
                    }
                    preBlockIdx = j;
                }
            }
        }

        System.out.println(answer);
    }
}

나의 풀이 (열 기준)

• 각 열마다 왼쪽과 오른쪽의 가장 높은 블록 중 작은 값만큼 물이 찰 수 있다고 보고, 현재 높이를 뺀 만큼을 빗물로 계산

import java.io.*;
import java.util.*;

/**
 * 백준 14719 빗물 (골드 5) - 구현
 * <a href="https://www.acmicpc.net/problem/14719">...</a>
 */
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        st = new StringTokenizer(br.readLine());

        int H = Integer.parseInt(st.nextToken());
        int W = Integer.parseInt(st.nextToken());

        int[] height = new int[W];

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

        int answer = 0;
        for (int i = 1; i < W - 1; i++) {
            int curHeight = height[i];

            int left = 0;
            for (int j = 0; j < i; j++) {
                left = Math.max(left, height[j]);
            }

            int right = 0;
            for (int j = i + 1; j < W; j++) {
                right = Math.max(right, height[j]);
            }

            if (curHeight < left && curHeight < right) answer += Math.min(left, right) - curHeight;

        }

        System.out.println(answer);
    }
}