알고리즘 스위치

2024. 4. 3. 09:36카테고리 없음

import java.io.*;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int switchNum = Integer.parseInt(br.readLine());

        StringTokenizer st = new StringTokenizer(br.readLine());

        int[] switchArr = new int[switchNum];

        for (int i = 0; i < switchNum; i++) {
            switchArr[i] = Integer.parseInt(st.nextToken());
        }

        int studentNum = Integer.parseInt(br.readLine());

        for (int i = 0; i < studentNum; i++) {
            st = new StringTokenizer(br.readLine());

            int gender = Integer.parseInt(st.nextToken());

            int n = Integer.parseInt(st.nextToken());

            if (gender == 1) {
                for (int j = n; j <= switchNum; j += n) {
                    switchArr[j - 1] ^= 1;
                }
            } else {
                n--;
                switchArr[n] ^= 1;
                int left = n - 1;
                int right = n + 1;
                while (left >= 0 && right < switchNum && switchArr[left] == switchArr[right]) {
                    switchArr[left] ^= 1;
                    switchArr[right] ^= 1;
                    left--;
                    right++;
                }
            }
        }
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < switchNum; i++) {
            sb.append(switchArr[i]).append(" ");
            if ((i + 1) % 20 == 0 || i == switchNum - 1) {
                sb.append("\n");
            }
        }
        System.out.println(sb);
        br.close();
    }
}

처음에 계속 이유없이 오답이었는데 범위설정에 문제가 있었다.

이전 코드에서는 여학생 부분에서 J<swtichNum으로 설정해서 스위치 번호가 배열의 인덱스보다 1크기때문에 마지막 스위치까지 처리를 못했다