백준 12927 배수 스위치 문제풀이 (JAVA)

문제 링크

문제

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

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

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

입력

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

출력

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

풀이

아까 킨거 또 끄기 싫으니 왼쪽부터 끄기 시작한다.
왼쪽부터 순차적으로 돌면서, 만약 꺼져있으면 키고, 킨 전구의 배수를 전부 반전시킨다.

소스코드

import java.util.*;
import java.io.*;
public class Main {

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();

        final String INPUT = br.readLine();
        final int NUMBER_OF_LIGHT = INPUT.length();

        boolean light[] = new boolean[NUMBER_OF_LIGHT + 1];
        for (int i = 1; i <= NUMBER_OF_LIGHT; i++) {
            if (INPUT.charAt(i - 1) == 'Y') {
                light[i] = true;
            }
        }

        int answer = 0;
        for (int i = 1; i <= NUMBER_OF_LIGHT; i++) {
            if (light[i]) {
                answer++;
                int k = i;
                while (k <= NUMBER_OF_LIGHT) {
                    light[k] = !light[k];
                    k += i;
                }
            }
        }

        sb.append(answer);
        sb.append("\n");

        bw.write(sb.toString());

        bw.flush();
        br.close();
        bw.close();

    }


}