[BOJ] 백준 1244 스위치 켜고 끄기 (KOI 2000 초등부)

문제 링크

www.acmicpc.net/problem/1244

1244번: 스위치 켜고 끄기

 

풀이

단순 시뮬레이션 문제입니다.

들어온 입력에 따라 문제에서 요구하는대로 구현하면 됩니다

스위치를 켜고, 끄는 연산은 1과 XOR해주면 쉽게 처리됩니다.

 

 

정답 코드

#include <bits/stdc++.h>

using namespace std;

int n, m;
int p[111];

int main() {
    cin >> n;
    for (int i = 1; i < n + 1; i++) {
        cin >> p[i];
    }
    cin >> m;
    while (m--) {
        int a, b;
        cin >> a >> b;
        if (a == 1) {
            for (int i = b; i < n + 1; i += b) {
                p[i] ^= 1;
            }
        } else {
            for (int i = 0; b + i < n + 1 && b - i > 0 && p[b + i] == p[b - i]; i++) {
                p[b - i] = (p[b + i] ^= 1);
            }
        }
    }
    for (int i = 1; i < n + 1; i++) {
        cout << p[i] << ' ';
        if (i % 20 == 0) cout << '\n';
    }
}