Node.js) 백준 1501번: 영어 읽기

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

1501번: 영어 읽기

const input = require('fs').readFileSync('./dev/stdin').toString().trim().split('\n');
const N = +input.shift();
const dict = new Map();
for (let i = 0; i < N; i++) {
	const word = input.shift().trim().split('');
	const headTail = word.shift() + (word.pop() ?? '');
	const body = word.sort().join('');

	if (dict.has(headTail)) {
		const subdict = dict.get(headTail);
		if (subdict.has(body)) {
			const cnt = subdict.get(body);
			subdict.set(body, cnt + 1);
		} else {
			subdict.set(body, 1);
		}
	} else {
		const subdict = new Map();
		subdict.set(body, 1);
		dict.set(headTail, subdict);
	}
}
const answer = [];
const M = +input.shift();
for (let i = 0; i < M; i++) {
	let tcAnswer = 0;
	const words = input.shift().trim();
	words.split(' ').forEach((word) => {
		word = word.split('');
		const headTail = word.shift() + (word.pop() ?? '');
		const body = word.sort().join('');
		if (dict.has(headTail)) {
			const subdict = dict.get(headTail);
			if (subdict.has(body)) {
				if (tcAnswer == 0) tcAnswer = 1;
				tcAnswer *= subdict.get(body);
			}
		}
	});
	answer.push(tcAnswer);
}
console.log(answer.join('\n'));