[백준] 14719 빗물 JavaScript

·2025년 2월 15일

문제

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

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

입력

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

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

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

출력

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

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

예제 입력

4 4
3 0 1 4

예제 출력

5

내가 했던 풀이 방법

  1. left와 right를 각각 0과 W-1로 설정하고 해당 인덱스에 해당하는 값으로 각 방향의 max값을 설정해준다.
  2. left가 right가 같아지기 전까지 다음 내용을 반복한다. 1) leftMax가 rightMax보다 작을 경우 left를 1 증가시키고 leftMax를 더 큰 값으로 갱신한다. 갱신된 leftMax를 기준으로 height[left]를 뺀 값을 더해준다. 이때 음수가 되는 것을 방지하고자 0과 max 값을 비교해주었다. 2) leftMax가 rightMax보다 크거나 같을 경우에는 right를 1 감소시키고 rightMax를 더 큰 값으로 갱신하고 rightMax에서 height[right]를 뺀 값을 더해준다.
  3. 이를 반복하면 totlaWater가 나오게 된다.

코드

const fs = require('fs');
let [n, height] = fs.readFileSync(0, 'utf-8').toString().trim().split('\n');

let [H, W] = n.split(' ').map(Number);
height = height.split(' ').map(Number);

let left = 0,
  right = W - 1;
let leftMax = height[left],
  rightMax = height[right];
let totalWater = 0;

while (left < right) {
  if (leftMax < rightMax) {
    left++;
    leftMax = Math.max(leftMax, height[left]);
    totalWater += Math.max(0, leftMax - height[left]);
  } else {
    right--;
    rightMax = Math.max(rightMax, height[right]);
    totalWater += Math.max(0, rightMax - height[right]);
  }
}

console.log(totalWater);

회고

왼쪽 방향에서 더 크거나 같은 height가 나올 때 사잇값을 처리해주는 방식으로 생각했는데 양방향에서 접근했으면 더 쉽게 풀 수 있었던 문제... 물론 위 방법으로는 예외가 많아서 70%정도만 맞더라... 투포인터 풀이 방법의 새로운 접근 법인 것 같다.

profile
Frontend🍓

0개의 댓글