취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 1244번 문제인 스위치 켜고 끄기이다.
문제는 아래 링크를 확인하자.

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

문제에서 주어진 것을 그대로 시뮬레이션하면 풀리는 문제이다. 그대로 구현하자.

0^1 = 1, 1^0 = 0이므로 xor 연산을 이용하면 간단하게 구현할 수 있다.

여학생의 조건을 구현할 때, 배열을 벗어나면 안 된다는 점을 신경쓰는 것이 좋다.

글쓴이는 배열의 양 끝에 -1과 999라는 있을 수 없는 값을 넣어 비교값이 거짓이 나오게 하였다.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;

int arr[102];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	int N; cin >> N;
	for (int i = 1; i <= N; i++) {
		cin >> arr[i];
	}
	
	arr[0] = -1, arr[N + 1] = 999;

	int Q; cin >> Q;
	while (Q--) {
		int x, k; cin >> x >> k;
		if (x == 1) {
			for (int i = k; i <= N; i += k) {
				arr[i] ^= 1;
			}
		}
		else {
			int dist = 1;
			arr[k] ^= 1;
			while (arr[k - dist] == arr[k + dist]) {
				arr[k - dist] ^= 1;
				arr[k + dist] ^= 1;
				dist++;
			}
		}
	}
	for (int i = 1; i <= N; i++) {
		cout << arr[i] << ' ';
		if (i % 20 == 0) cout << '\n';
	}
}