백준[JS] > 1244번 스위치 켜고 끄기

문제링크

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

문제풀이

스위치를 어레이형태로 표현해서 이를 다루었다. switchArr[n] 은 n번째 스윗치의 상태를 알려준다 ( 1 또는 0 ) 이 이후에는 구현능력에 달려있다. 어렵지 않게 풀 수 있었다.

let fs = require("fs");
let input = fs.readFileSync("/dev/stdin").toString().trim().split("\n");
let switchCount = Number(input[0]);

let switchArr = input[1].split(" ").map((n) => Number(n));
switchArr.unshift(undefined);
for (let i = 3; i < input.length; ++i) {
    let [gender, n] = input[i].split(" ").map((v) => Number(v));

    if (gender === 1) {
        // 남자
        let multipliedNum = n;
        while (multipliedNum <= switchCount) {
            switchArr[multipliedNum] = switchArr[multipliedNum] === 1 ? 0 : 1;
            multipliedNum += n;
        }
    } else {
        // 여자
        switchArr[n] = switchArr[n] === 0 ? 1 : 0;
        let left = n - 1,
            right = n + 1;

        while (left >= 1 && right <= switchCount && switchArr[left] === switchArr[right]) {
            switchArr[left] = switchArr[left] === 0 ? 1 : 0;
            switchArr[right] = switchArr[right] === 0 ? 1 : 0;
            left -= 1;
            right += 1;
        }
    }
}

switchArr.shift();
let answer = "";
let cnt = 0;
for (let i = 0; i < switchArr.length; ++i) {
    if (cnt !== 20) {
        answer += ` ${switchArr[i]}`;
        ++cnt;
    } else {
        answer += `\n${switchArr[i]}`;
        cnt = 1;
    }
}
console.log(answer.trim());