[백준 1620] 나는야 포켓몬 마스터 이다솜 with Node.js

문제 링크

www.acmicpc.net/problem/1620

1620번: 나는야 포켓몬 마스터 이다솜

풀이

const input = require("fs").readFileSync("/dev/stdin").toString().trim().split(/\s/);
const n = +input[0];
const m = +input[1];
const arr = input.slice(2, n+2);
const pokemonMap = new Map(arr.map((v, i) => [v, i+1]));
const question = input.slice(n+2);
const answer = [];
question.forEach(v => {
    if (Number.isNaN(+v)) answer.push(pokemonMap.get(v));
    else answer.push(arr[+v-1]);
});
console.log(answer.join("\n"));

포켓몬 이름을 저장한 배열과, 그 배열로 포켓몬 이름과 번호를 키와 값으로 갖는 Map 객체를 준비한다. 문제가 문자열이면 Map객체에서 번호를 찾고, 숫자면 배열에서 포켓몬 이름을 찾는다.

 

무언가 깔끔하진 않지만, Map객체 없이 배열만으로 번호를 찾으려면 arr.indexOf를 사용해야 하는데, 이러면 시간 초과가 뜬다.