Node.js) 백준 28683번: 피타! 피타! 피타츄!

 

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

28683번: 피타! 피타! 피타츄!

const n = +require('fs').readFileSync('./dev/stdin').toString();

function isSquareNumber(number) {
	const x = Math.sqrt(number);
	return Math.floor(x) == x;
}

let answer = 0;
if (isSquareNumber(n)) {
	//
	answer = -1;
} else {
	// 나머지 두 변이 모두 정수이어야 함.
	//case 1: n 이 빗변인 경우
	const possible = new Set();
	for (let a = 1; a * a < n; a++) {
		possible.add(a * a);
	}

	for (const p of possible) {
		const a = p;
		possible.delete(p);
		if (n == a * 2) {
			answer++;
		} else if (possible.has(n - a)) {
			possible.delete(n - a);
			answer++;
		}
	}

	//case 2:  n이 빗변이 아닌 경우
	let a = 1;
	while (true) {
		if ((n - (Math.pow(a + 1, 2) - 1)) % (a * 2) == 0) {
			answer++;
		}
		if (Math.pow(a + 1, 2) - 1 > n) {
			break;
		}
		a++;
	}
}
console.log(answer);