[JavaScript] 백준 코딩테스트 10810. 브론즈 3 : 공 넣기

📌  문제

10810번: 공 넣기

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.

도현이는 앞으로 M번 공을 넣으려고 한다. 도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다. 만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다.

공을 어떻게 넣을지가 주어졌을 때, M번 공을 넣은 이후에 각 바구니에 어떤 공이 들어 있는지 구하는 프로그램을 작성하시오.

📌  입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 넣는 방법이 주어진다. 각 방법은 세 정수 i j k로 이루어져 있으며, i번 바구니부터 j번 바구니까지에 k번 번호가 적혀져 있는 공을 넣는다는 뜻이다. 예를 들어, 2 5 6은 2번 바구니부터 5번 바구니까지에 6번 공을 넣는다는 뜻이다. (1 ≤ i ≤ j ≤ N, 1 ≤ k ≤ N)

도현이는 입력으로 주어진 순서대로 공을 넣는다.

📌  출력

1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 공이 들어있지 않은 바구니는 0을 출력한다.

📌  예제 입출력

---

📌  풀이

알아야 하는 것

• Map 객체

구할 것

• 데이터 가공
  • 첫 줄에서 바구니 개수와 공을 넣을 횟수가 주어진다. 후자의 경우 쓸 일이 없어서 바구니 개수만 상수 basket에 할당한다.
  • 나머지 입력값은 [i, j, k] 형태로, i번 바구니부터 j번 바구니까지 k번 번호의 공을 넣는다는 뜻이다.
• Map 객체
  • Map 객체는 어떤 키를 가진 값이 이미 존재할 때, 그 키에 새로운 값을 넣으면 갱신이 된다.
  • 문제에서 공이 이미 들어 있다면 그 공을 빼고 새로 넣으라고 하는데, Map 객체를 사용하면 공을 빼고 넣을 로직을 고려하지 않아도 새로운 공이 들어간다.
  • 예컨대 basket = 5라면 바구니를 총 5개 가지고 있어서, 1번 ~ 5번 바구니가 map의 키로 설정되어야 한다. 따라서 for문을 돌 때 i = 1로 시작하고, 가장 처음에 바구니에는 공이 들어 있지 않기 때문에 모든 키의 값을 0으로 초기화한다.
  • 위에서 말한 나머지 입력값 (count) = [i, j, k] 형태로 i와 j는 범위다. 예컨대 1 4 라고 한다면 1번 바구니, 2번 바구니, ... 4번 바구니까지 포함되어야 한다. 따라서 for문으로 증가시킨 바구니에 k번 번호의 공을 밸류값으로 set한다. 
  • 구해야 하는 것은 바구니에 들어 있는 공의 번호, 즉 map의 밸류값이다. 따라서 map.values()로 밸류값만 순회해 answer에 push한다.
  • 출력 시 배열이 아닌 '공의 번호를 공백으로 구분한' 문자열이므로 join을 써서 출력한다.

const fs = require("fs");
const input = fs.readFileSync("/dev/stdin").toString().trim().split("\n");

const basket = input.shift().split(" ").map(Number)[0];
const count = input.map((v) => v.split(" ").map(Number));

const map = new Map();

for (let i = 1; i <= basket; i++) {
  map.set(i, 0);
}

for (let i = 0; i < count.length; i++) {
  for (let j = count[i][0]; j <= count[i][1]; j++) {
    map.set(j, count[i][2]);
  }
}

const answer = [];

for (let ball of map.values()) {
  answer.push(ball);
}

console.log(answer.join(" "));