흙금이네 블로그

[BOJ] 21313 - 문어 (Python, JavaScript) 본문

알고리즘

[BOJ] 21313 - 문어 (Python, JavaScript)

흙금 2023. 2. 14. 23:59

 

 

아이디어

 

사전 순으로 가장 앞서는 수열이 되기 위한 규칙을 찾는다.

 

 

풀이 #1 (Python)

 

문어들이 1번과 2번 손을 번갈아 잡는 수열이 사전 순으로 가장 앞선다.

그러나 문어의 수가 홀수인 경우 1번과 2번 손만 사용해서는 원을 만들 수 없으므로 3번 손도 사용한다.

따라서 문어 수를 2로 나눠 소수점 이하를 버린 값만큼 1과 2가 담긴 리스트를 반복하고, 홀수면 3도 추가해 언패킹한다.

 

N = int(input())
print(*[1, 2]*(N//2)+[3]*(N%2))

 

 

문자열로 구현할 수도 있는데, 짝수인 경우 끝에 공백이 생기므로 rstrip 메서드로 공백을 제거한다.

print(('1 2 '*(N//2)+'3'*(N%2)).rstrip())

 

 

 

풀이 #2 (JavaScript)

 

입력으로 주어지는 문어는 최소 4마리이므로, res에 공백 없이 1과 2를 저장하고 시작한다.

문어 수 절반에 소수점 이하를 버린 값보다 1 작게 for문을 돌며 1과 2를 추가하고, 홀수인 경우 끝에 3을 추가한다.

 

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

const N = +input[0];
let res = '1 2';
for (let i=1; i<parseInt(N/2); i++) res += ' 1 2';
if (N%2) res += ' 3';
console.log(res);

 

Comments