BOJ 1244 스위치 켜고 끄기 JAVA

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

1244번: 스위치 켜고 끄기

 

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

public class 스위치켜고끄기 {
	
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		int switches[] = new int[N+1];
		StringTokenizer st =new StringTokenizer(br.readLine());
		for(int i=1; i<=N; i++) {
			switches[i] = Integer.parseInt(st.nextToken()); 
		}
		//System.out.println(Arrays.toString(switches));
		int studentNum = Integer.parseInt(br.readLine());
		for(int i=0; i<studentNum; i++) {
			st = new StringTokenizer(br.readLine());
			
			int student = Integer.parseInt(st.nextToken());
			int num = Integer.parseInt(st.nextToken());
			
			if(student==1) {
				for(int j=1; j<=N; j++) {
					if(j%(num)==0) {
						if(switches[j] == 0) {
							
							switches[j] = 1;
						}
						else {
							switches[j] = 0;
						}
						
					}
				}
			}
			//System.out.println(Arrays.toString(switches)); 
			else {
				int startIdx = num;
				int left = startIdx-1;
				int right = startIdx+1;
				
				
				
				while(left>=1 && right<=N) {
					
					if(switches[left]==switches[right]) {
						if(switches[left] == 0 ) {
							switches[left]=1;
							switches[right]=1;
						}
						else if(switches[left] ==1 ) {
							switches[left]=0;
							switches[right]=0;
						}
						
						left--;
						right++;
						
					}
						
					
					
				}
				if(switches[startIdx] ==0) {
					switches[startIdx]=1;
				}
				else {
					switches[startIdx]=0;
				}
				
			}
		
		}
		
		for(int i=1; i<=N; i++) {
			System.out.print(switches[i]+" ");
			if(i%20==0)System.out.println();
		}
	}
}