[백준/Rust]14719번 : 빗물

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

---

문제

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

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

입력

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

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

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

출력

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

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

예제 입출력

---

풀이

전형적인, 그리고 대표적인 투포인터 문제라 할 수 있다.

---

코드

use std::io;

fn calc(w: usize, h: Vec<usize>) -> usize {
    if h.is_empty() || h.len() < 3 {
        return 0;
    }
    
    let mut l = 0;
    let mut r = w - 1;
    let mut l_max = h[l];
    let mut r_max = h[r];
    let mut water = 0;
    
    while l < r {
        if l_max < r_max {
            l += 1;
            l_max = l_max.max(h[l]);
            water += l_max.saturating_sub(h[l]);
        } else {
            r -= 1;
            r_max = r_max.max(h[r]);
            water += r_max.saturating_sub(h[r]);
        }
    }
    
    water
}

fn main() {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let dims: Vec<usize> = input.trim().split_whitespace()
        .map(|x| x.parse().unwrap())
        .collect();
    let _h = dims[0];
    let w = dims[1];

    input.clear();
    io::stdin().read_line(&mut input).unwrap();
    let heights: Vec<usize> = input.trim().split_whitespace()
        .map(|x| x.parse().unwrap())
        .collect();

    let res = calc(w, heights);
    println!("{}", res);
}