[BOJ] 14719 - 빗물 JAVA

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:

힌트 2:

힌트 3:

 

 

 

 

 

 

 

 

풀이 .

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

class Block {
    int w, h;
    Block(int w, int h) {
        this.w = w;
        this.h = h;
    }
}

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());
        Block[] arr = new Block[W];
        int sumOfBlock = 0;

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

        // 배열의 시작부터 오름차순 블록만 받아서 저장
        Stack<Block> leftStk = new Stack<>();
        Stack<Block> rightStk = new Stack<>();
        int maxHeight = 0;
        for(int i = 0; i < arr.length; i++) {
            if(arr[i].h > maxHeight) {
                maxHeight = arr[i].h;
                leftStk.push(arr[i]);
            }
        }

        // 배열의 끝부터 오름차순 블록만 받아서 저장
        maxHeight = 0;
        for(int i = arr.length - 1; i >= 0; i--) {
            if(arr[i].h > maxHeight) {
                maxHeight = arr[i].h;
                rightStk.push(arr[i]);
            }
        }

        // 모든 면적 계산
        int ans = (rightStk.peek().w - leftStk.peek().w + 1) * maxHeight;
        int beforeW = leftStk.pop().w;
        while(!leftStk.isEmpty()) {
            int w = leftStk.peek().w;
            int h = leftStk.peek().h;
            ans += (beforeW - w) * h;
            beforeW = w;
            leftStk.pop();
        }
        beforeW = rightStk.pop().w;
        while(!rightStk.isEmpty()) {
            int w = rightStk.peek().w;
            int h = rightStk.peek().h;
            ans += (w - beforeW) * h;
            beforeW = w;
            rightStk.pop();
        }

        // 전체 면적에서 블록 면적 제거
        ans -= sumOfBlock;
        System.out.println(ans);
    }
}

 

codeung.tistory.com/253 와 똑같은 문제.

 

스택을 이용하여 해결했지만 굳이 스택일 필요는 없다. 어레이리스트나 최대힙도 가능.