Node.js) 백준 1917번: 정육면체 전개도

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

1917번: 정육면체 전개도

const input = require('fs')
	.readFileSync('./dev/stdin')
	.toString()
	.trim()
	.split('\n')
	.map((v) => v.split(' ').map(Number));
const samples = [
	[
		[1, 0, 0, 0],
		[1, 1, 1, 1],
		[1, 0, 0, 0],
	],
	[
		[0, 1, 0, 0],
		[1, 1, 1, 1],
		[1, 0, 0, 0],
	],
	[
		[1, 0, 0, 0],
		[1, 1, 1, 1],
		[0, 1, 0, 0],
	],
	[
		[0, 0, 1, 0],
		[1, 1, 1, 1],
		[1, 0, 0, 0],
	],
	[
		[1, 0, 0, 0],
		[1, 1, 1, 1],
		[0, 0, 1, 0],
	],
	[
		[1, 0, 0, 0],
		[1, 1, 1, 1],
		[0, 0, 0, 1],
	],
	[
		[0, 0, 0, 1],
		[1, 1, 1, 1],
		[1, 0, 0, 0],
	],
	[
		[0, 1, 0, 0],
		[1, 1, 1, 1],
		[0, 1, 0, 0],
	],
	[
		[0, 0, 1, 0],
		[1, 1, 1, 1],
		[0, 1, 0, 0],
	],
	[
		[0, 1, 0, 0],
		[1, 1, 1, 1],
		[0, 0, 1, 0],
	],
	[
		[0, 0, 1, 1, 1],
		[1, 1, 1, 0, 0],
	],
	[
		[1, 1, 1, 0, 0],
		[0, 0, 1, 1, 1],
	],
	[
		[0, 0, 1, 1],
		[0, 1, 1, 0],
		[1, 1, 0, 0],
	],
	[
		[1, 1, 0, 0],
		[0, 1, 1, 0],
		[0, 0, 1, 1],
	],
	[
		[0, 0, 1, 1],
		[1, 1, 1, 0],
		[1, 0, 0, 0],
	],
	[
		[1, 0, 0, 0],
		[1, 1, 1, 0],
		[0, 0, 1, 1],
	],
	[
		[1, 1, 0, 0],
		[0, 1, 1, 1],
		[0, 1, 0, 0],
	],
	[
		[0, 1, 0, 0],
		[0, 1, 1, 1],
		[1, 1, 0, 0],
	],
	[
		[0, 1, 0, 0],
		[1, 1, 1, 0],
		[0, 0, 1, 1],
	],
	[
		[0, 0, 1, 1],
		[1, 1, 1, 0],
		[0, 1, 0, 0],
	],
]
	.map((v) => rotate(v))
	.flat();

function rotate(arr1) {
	const r = arr1.length;
	const c = arr1[0].length;

	const arr2 = Array.from(Array(c), () => Array(r).fill(0));
	for (let i = 0; i < r; i++) {
		for (let j = 0; j < c; j++) {
			arr2[j][r - i - 1] = arr1[i][j];
		}
	}

	const arr3 = Array.from(Array(r), () => Array(c));
	for (let i = 0; i < r; i++) {
		for (let j = 0; j < c; j++) {
			arr3[r - i - 1][c - j - 1] = arr1[i][j];
		}
	}
	const arr4 = Array.from(Array(c), () => Array(r));
	for (let i = 0; i < r; i++) {
		for (let j = 0; j < c; j++) {
			arr4[c - j - 1][i] = arr1[i][j];
		}
	}
	return [arr1, arr2, arr3, arr4];
}

const answer = [];
let board = [];
for (let t = 0; t < 3; t++) {
	board = [];
	for (let i = 0; i < 6; i++) {
		board.push(input.shift());
	}

	for (let s = 0; s < samples.length; s++) {
		const sample = samples[s];

		if (check(sample)) {
			answer.push('yes');
			break;
		}
	}
	if (answer.length == t) {
		answer.push('no');
	}
}

console.log(answer.join('\n'));

function check(arr) {
	const r = arr.length;
	const c = arr[0].length;
	for (let i = 0; i <= 6 - r; i++) {
		for (let j = 0; j <= 6 - c; j++) {
			if (checkUnit(i, j, arr)) return true;
		}
	}
	return false;
}

function checkUnit(x, y, arr) {
	const r = arr.length;
	const c = arr[0].length;
	for (let i = 0; i < r; i++) {
		for (let j = 0; j < c; j++) {
			if (board[i + x][j + y] !== arr[i][j]) {
				return false;
			}
		}
	}
	return true;
}