백준 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

---

푸는데 꽤 오래 걸렸던 문제.. 스택을 사용했다.

풀이

1. idx 변수에 현재까지 가장 높은 높이를 저장한다.

2. idx보다 작거나 같은 수가 들어오면 (들어온 수 = now)

 - 스택 peek값이 자신보다 작은 수일 때

peek 값이 자신보다 같거나 큰 수가 나올 때까지 pop 해준 후 now - stack.peek()를 answer에 더해준다. pop 해준만큼 now 값을 다시 스택에 push한다. -> 만약 413이었다면 answer에 2를 더해주고 그 부분이 칠해졌다고 생각한다. 즉, 433이 되었다고 생각한다. 

- peek 값이 자신보다 크다면 그냥 아무것도 하지 않는다

- 그 후 now 값을 스택에 push

3. idx보다 큰 수가 들어오면(들어온 수 = now)

: stack에 있는 값을 모두 pop해주며 answer += idx - stack.pop() 해준다.

그 후 idx = now를 해준다.

 

import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int h = sc.nextInt();
        int w = sc.nextInt();
        int idx = sc.nextInt();
        int answer = 0;
        Stack<Integer> stack = new Stack<>();
        for (int i = 1; i < w; i++) {
            int now = sc.nextInt();
            int cnt = 0;
            if (idx >= now) {   
                while (!stack.isEmpty() && stack.peek() < now) {
                    int tmp = stack.pop();
                    answer += now - tmp;
                    cnt++;
                }
                while (cnt--> 0) {
                    stack.push(now);
                }
                stack.push(now);
            }
            else {
                while (!stack.isEmpty()) {
                    answer += idx - stack.pop();
                }
                idx = now;
            }

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