BOJ_S3_1244

📁 [S3_1244] 스위치 켜고 끄기

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

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

		int[] arr = new int[sw];
		
		StringTokenizer st = new StringTokenizer(br.readLine(), " ");
		// 스위치 번호가 1부터 시작 ( 편의를 위해 0번째는 비워둔다 )
		for (int i = 0; i < sw; i++) {
			arr[i] = Integer.parseInt(st.nextToken());
		}

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

		for (int i = 0; i < stu_cnt; i++) {
			st = new StringTokenizer(br.readLine(), " ");
			int sex = Integer.parseInt(st.nextToken());
			int num = Integer.parseInt(st.nextToken());
			
			// 남자
			if (sex == 1) {
				for (int z = 0; z < sw; z++) {
					if ((z+1) % (num) == 0) {
						arr[z] = (arr[z] == 0) ? 1 : 0;

					} else {
						continue;
					}
				}
			}

			// 여자
			else if (sex == 2) {
				// 처음에 자기자신 바꿔주기
				arr[num-1] = (arr[num-1] == 0) ? 1 : 0;
				int left = num-1 - 1;
				int right = num-1 + 1;
				// 왼,오 값이 다르거나, 왼쪽, 오른쪽이 범위를 벗어날 때 끝내기
				while (true) {
					if (left < 0 || right > sw-1 || arr[left] != arr[right] ) {
						break;
					} else {
						arr[left] = (arr[left] == 0) ? 1 : 0;
						arr[right] = (arr[right] == 0) ? 1 : 0;
						left--;
						right++;
					}
				}
			}
		}
		for (int a = 0; a < sw; a++) {
			System.out.print(arr[a] + " ");
			if((a+1) % 20 == 0) {
				System.out.println();
			}
		}

	}
}     

🤔 나의 생각

일단 삼항연산자를 사용하여 숫자 바꿔주기를 하였다.
그리고 남자는 간단하게 배수 찾기로 하였고
여자는 범위를 벗어나는 경우 말고 무한루프문을 통해 왼쪽, 오른쪽을 비교해 주었다.
처음에 배열을 1 index부터 주어서 풀었는데 계속해서 런타임에러(ArrayIndexOutOfBounds)가 떴다.
그래서 0 index 부터 주니까 문제가 해결되었다.
이 문제에서는 수를 바꾸는 것도 바꾸는 거지만 스위치 순서가 1부터라서 배열과 매치하는게 가장 헷갈렸던 것 같다.