백준 1244번 스위치 켜고 끄기 C++

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

 

입력받은 순서대로 시뮬레이션 해가면서 문제를 해결하였다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>

using namespace std;
using ll = long long;

void ChangeSwitch(int gender, int num, vector<bool>& sw) {
	if (gender == 1) {
		int idx = num;
		while (idx < sw.size()) {
			sw[idx] = !sw[idx];
			idx += num;
		}
	}
	else {
		sw[num] = !sw[num];
		int left = num - 1;
		int right = num + 1;
		while (true) {
			if (left < 1 || right >= sw.size()) break;
			if (sw[left] != sw[right]) break;
			sw[left] = !sw[left];
			sw[right] = !sw[right];
			left--;
			right++;
		}
	}
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int n, m;
	cin >> n;
	vector<bool> sw(n + 1, false);
	for (int i = 1; i <= n; i++) {
		int input;
		cin >> input;
		if (input == 1) sw[i] = true;
	}
	cin >> m;
	for (int i = 0; i < m; i++) {
		int gender, num;
		cin >> gender >> num;
		ChangeSwitch(gender, num, sw);
	}
	for (int i = 1; i <= n; i++) {
		cout << sw[i] << " ";
		if (i % 20 == 0) cout << "\n";
	}
};