[BOJ] 21918 - 전구 (Python, JavaScript)

 

 

아이디어

 

명령어에 따라 전구 상태를 변경한다.

 

 

풀이 #1 (Python)

 

입력된 명령어에 따라 전구 상태 리스트 bulb의 상태 값을 슬라이싱을 이용해 변경한다.

 

import sys

input = sys.stdin.readline

N, M = map(int, input().split())
bulb = list(map(int, input().split()))
for _ in range(M):
    a, b, c = map(int, input().split())
    if a == 1:
        bulb[b-1] = c
    elif a == 2:
        bulb[b-1:c] = [1-i for i in bulb[b-1:c]]
    elif a == 3:
        bulb[b-1:c] = [0]*(c-b+1)
    else:
        bulb[b-1:c] = [1]*(c-b+1)
print(*bulb)

 

 

for문으로 리스트의 값을 변경하는 코드보다 슬라이싱을 이용해 변경하는 코드가 훨씬 빨랐다.

# 더 느린 코드
elif a == 2:
    for i in range(b-1, c):
        bulb[i] = 1-bulb[i]
elif a == 3:
    for i in range(b-1, c):
        bulb[i] = 0
else:
    for i in range(b-1, c):
        bulb[i] = 1

 

 

 

풀이 #2 (JavaScript)

 

입력된 명령어에 따라 전구 상태 리스트 bulb의 상태 값을 for문을 이용해 변경한다.

 

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

let value = input;
let [N, M] = value[0].split(' ').map(Number);
let bulb = value[1].split(' ').map(Number);
for (let i=2; i<M+2; i++) {
    let [a, b, c] = value[i].split(' ').map(Number);
    if (a === 1) bulb[b-1] = c;
    else if (a === 2) {
        for (let j=b-1; j<c; j++) bulb[j] = 1-bulb[j];
    }
    else if (a === 3) {
        for (let j=b-1; j<c; j++) bulb[j] = 0;
    }
    else {
        for (let j=b-1; j<c; j++) bulb[j] = 1;
    }
}
console.log(bulb.join(' '));

 

 

스프레드 연산자를 사용하는 코드보다 join 메서드를 사용하는 코드가 메모리가 더 적게 사용되고 더 빨랐다.

// 1.
console.log(...bulb);

// 2. 메모리가 더 적게 사용되고 더 빠른 코드
console.log(bulb.join(' '));

 

문자를 숫자로 바꾸기 위해 map 메서드에서 화살표 없이 Number 함수명만 작성한 코드가 다른 두 코드보다 더 빨랐다.

// 1.
let [a, b, c] = value[i].split(' ').map(val => parseInt(val));

// 2.
let [a, b, c] = value[i].split(' ').map(val => Number(val));

// 2. 더 빠른 코드
let [a, b, c] = value[i].split(' ').map(Number);