[Baekjoon]12927번 배수 스위치 - Javascript

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

12927번: 배수 스위치

문제

강호는 전구 N개를 가지고 있다. 전구는 1번부터 N번까지 번호가 매겨져 있으며, 일렬로 놓여져 있다. 전구는 켜져있거나 꺼져있다.

강호는 모든 전구를 끄려고 한다. 강호는 전구를 켜고 끌 수 있는 스위치 N개를 가지고 있고, 스위치도 1번부터 N번까지 번호가 매겨져 있다. i번 스위치는 i의 배수 번호를 가지는 전구의 상태를 모두 반전시킨다.

현재 전구의 상태가 주어졌을 때, 모든 전구를 끄기 위해서 스위치를 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 전구의 상태가 1번 전구부터 차례대로 주어진다. Y는 전구가 켜 있는 경우, N은 전구가 꺼져있는 경우이다. 전구의 개수는 1보다 크거나 같고 1,000보다 작거나 같은 자연수이다.

출력

모든 전구를 끄기 위해서 스위치를 몇 번 눌러야 하는지 출력한다. 만약, 모든 전구를 끌 수 없다면 -1을 출력한다.

예제 입력 1 복사

YYYYYY

예제 출력 1 복사

1

예제 입력 2 복사

YNYNYNYNY

예제 출력 2 복사

2

예제 입력 3 복사

NNNNNNNNNN

예제 출력 3 복사

0

예제 입력 4 복사

YYYNYYYNYYYNYYNYYYYN

예제 출력 4 복사

4

 

문제풀이(1)

원래는 에라토스테네스의 체를 생각했지만 i번째 이전의 값들은 수정하지 않는다는 것을 다시 깨닫고 풀었다.

1. i번째의 배수마다 전등의 동작 유무를 반전시키며 카운트를 해준다.
2. 모든 스위치가 꺼졌을 때의 카운트를 출력한다.

const filePath = process.platform === "linux" ? "/dev/stdin" : "./input.txt";
const arr = require("fs").readFileSync(filePath).toString().trim().split("");

let count = 0;

for (let i = 0; i < arr.length; i++) {
  if (arr[i] === "Y") {
    arr[i] = "N";
    count++;

    for (let j = i + 1; j <= arr.length; j += i + 1) {
      arr[j - 1] = arr[j - 1] === "Y" ? "N" : "Y";
    }
  }
}

console.log(count);