알고리즘
백준 1244번 스위치 켜고 끄기 C++
영춘권의달인
2022. 4. 17. 15:03
입력받은 순서대로 시뮬레이션 해가면서 문제를 해결하였다.
#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";
}
};