[백준] 14719번 : 빗물 - 자바(Java)

Problem 🔒

문제

https://www.acmicpc.net/problem/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:

힌트 2:

힌트 3:

---

Approach 1 ❌

1. 배열에서 제일 높은 위치들을 찾고 양쪽을 탐색하는 방법으로 접근
   • 탐색하면서 그 다음으로 높은 위치를 찾아 그 사이에 빈 칸들을 찾기

---

Approach 2 ⭕

1. 현재 블록 위치에서 왼쪽 블록들을 탐색해서 가장 높은 높이 구하기
2. 현재 블록 위치에서 오른쪽 블록들을 탐색해서 가장 높은 높이 구하기
3. 두 높이 중에서 낮은 높이와 현재 높이의 차이를 구하기

주의할 점

• 구한 왼쪽, 오른쪽 블록들의 높이가 현재 높이보다 높아야 한다.
• 첫 블럭과 끝 블럭은 빗물이 고일 수 없다.

---

Solution💡

import java.util.Scanner;

public class 빗물 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int H = sc.nextInt();
        int W = sc.nextInt();

        int[] arr = new int[W];
        for(int i=0; i<W; i++) arr[i] = sc.nextInt();

        int sum = 0;
        for(int i=0; i<W; i++){
            int left = 0;
            int right = 0;

            // 왼쪽으로
            // j는 i-1이 아닌 i부터 시작(첫 블럭 고려)
            for(int j=i; j>=0; j--){
                if(left == H) break;
                if(arr[j] > arr[i] && arr[j] > left) left = arr[j];
            }

            // 오른쪽으로
            // j는 i+1이 아닌 i부터 시작(끝 블럭 고려)
            for(int j=i; j<W; j++){
                if(right == H) break;
                if(arr[j] > arr[i] && arr[j] > right) right = arr[j];
            }

            if(left == 0 || right == 0) continue;
            sum += (left <= right) ? (left-arr[i]) : (right-arr[i]);

        }
        System.out.println(sum);
    }
}