[JavaScript][node.js] 백준 18111 마인크래프트

문제

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

18111번: 마인크래프트

자바스크립트(node.js)로 풀이했습니다.

최종코드

const fs = require('fs');
const filePath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';

const input = fs.readFileSync(filePath).toString().trim().split('\n');
const [n, m, b] = input[0].split(' ');

const soils = [];
input.slice(1).forEach((line) => {
  soils.push(...line.split(' ').map(Number));
});
const soilCounts = new Array(257).fill(0);
soils.forEach((soil) => (soilCounts[soil] += 1));

const checkTime = (h, inventory) => {
  let time = 0;
  for (const soil in soilCounts) {
    if (h > soil) {
      time += soilCounts[soil] * (h - soil);
      inventory -= soilCounts[soil] * (h - soil);
    } else if (h < soil) {
      time += 2 * soilCounts[soil] * (soil - h);
      inventory += soilCounts[soil] * (soil - h);
    }
  }
  if (inventory < 0) {
    return Infinity;
  }
  return time;
};

let minTime = Infinity;
let maxHeight = 0;
for (let height = 0; height <= 256; height++) {
  const finalTime = checkTime(height, b);

  if (finalTime < minTime) {
    minTime = finalTime;
    maxHeight = height;
  } else if (finalTime === minTime) {
    if (maxHeight < height) {
      maxHeight = height;
    }
  }
}

console.log(minTime, maxHeight);

인사이트

• 비둘기집의 원리 기억하자! 땅의 높이는 0 ~ 256 사이의 값이므로 500 * 500 순회를 256 순회로 줄일 수 있다.
• 배열의 총 길이를 알고 그 배열을 0으로 채우고 싶은 경우
  • new Array(N).fill(0);
• 배열의 각 요소를 number형으로 바꾼 새로운 배열을 얻고 싶은 경우
  • arr.map(Number);
• 배열은 const로 만들어도 값 변경 가능