Node.js) 백준 5052번: 전화번호 목록

 

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

5052번: 전화번호 목록

 

class TrieNode {
	constructor(character, isCompleteWord = false) {
		this.character = character;
		this.isCompleteWord = isCompleteWord;
		this.children = new Map();
	}

	getChild(character) {
		return this.children.get(character);
	}

	addChild(character, isCompleteWord = false) {
		if (!this.children.has(character)) {
			this.children.set(character, new TrieNode(character, isCompleteWord));
		}

		const childNode = this.children.get(character);

		childNode.isCompleteWord = childNode.isCompleteWord || isCompleteWord;

		return childNode;
	}

	hasChild(character) {
		return this.children.has(character);
	}

	hasChildren() {
		return this.children.size !== 0;
	}
}

class PhoneBook {
	constructor() {
		this.head = new TrieNode('*');
	}

	addPhoneNumber(numberString) {
		const characters = Array.from(numberString);
		let currentNode = this.head;

		for (let i = 0; i < characters.length; i++) {
			const isComplete = i == characters.length - 1;
			if (isComplete == true && currentNode.hasChild(characters[i])) {
				return false;
			}
			currentNode = currentNode.addChild(characters[i], isComplete);
			if (isComplete == false && currentNode.isCompleteWord == true) {
				return false;
			}
		}
		return true;
	}
}

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

let i = 0;
const T = +input[i++];
const answer = [];
for (let t = 0; t < T; t++) {
	const C = +input[i++];
	const phoneBook = new PhoneBook();
	let result = 'YES';
	for (let c = 0; c < C; c++) {
		const success = phoneBook.addPhoneNumber(input[c + i].trim());
		if (!success) {
			result = 'NO';
			break;
		}
	}
	answer.push(result);
	i = i + C;
}
console.log(answer.join('\n'));