알고리즘 스위치
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크기때문에 마지막 스위치까지 처리를 못했다